aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）: brand dna generate

Browse Source 
fix（修复bug）:
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zhouchengrong
2025-03-27 16:48:42 +08:00
parent 9fd911500b
commit 1deea783a1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/brand_dna/service_generate_brand_info.py
View File

@@ -41,7 +41,7 @@ class GenerateBrandInfo:
self.output_parser = StructuredOutputParser.from_response_schemas(self.response_schemas) self.output_parser = StructuredOutputParser.from_response_schemas(self.response_schemas)
self.format_instructions = self.output_parser.get_format_instructions() self.format_instructions = self.output_parser.get_format_instructions()
self.prompt = PromptTemplate( self.prompt = PromptTemplate(
template="你是一个时装品牌的设计师。根据用户输入提取出brand name，brand slogan,brand logo 描述。如果没有以上内容，需要你根据用户输入随意发挥。随后根据brand logo 描述生成一个prompt，这个prompt用于生成模型.\n{format_instructions}\n{question}", template="你是一个时装品牌的设计师。根据用户输入提取出brand name，brand slogan,brand logo 描述。如果没有以上内容，需要你根据用户输入随意发挥。随后根据brand logo 描述生成一个prompt，这个prompt用于生成模型，prompt需要完全表达用户的想法并使用英文，使用简洁明了的单词不要过长。.\n{format_instructions}\n{question}",
input_variables=["question"], input_variables=["question"],
partial_variables={"format_instructions": self.format_instructions} partial_variables={"format_instructions": self.format_instructions}
) )
@@ -98,7 +98,7 @@ class GenerateBrandInfo:
if __name__ == '__main__': if __name__ == '__main__':
request_data = GenerateBrandModel( request_data = GenerateBrandModel(
user_id="89", user_id="89",
prompt="xiaomi" prompt="华为"
) )
service = GenerateBrandInfo(request_data) service = GenerateBrandInfo(request_data)
print(service.get_result()) print(service.get_result())