ai/agent2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化prompt

Browse Source
This commit is contained in:
CounterFire2023 2024-12-16 22:50:18 +08:00
parent 5ee0dd9a19
commit ee55f9f23f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bot.py
View File

@ -26,7 +26,7 @@ if OLLAMA_BOT_TOKEN is None:
if __name__ == "__main__": if __name__ == "__main__":
application = ApplicationBuilder().token(OLLAMA_BOT_TOKEN).build() application = ApplicationBuilder().token(OLLAMA_BOT_TOKEN).build()
application.add_handler(chat_id.filter_handler, group=-1) # application.add_handler(chat_id.filter_handler, group=-1)
application.add_handlers([ application.add_handlers([
start_handler, start_handler,

6
llm/agent_util.py
View File

@ -43,7 +43,7 @@ def config_agent(llama3, llama3_json):
<|begin_of_text|> <|begin_of_text|>
<|start_header_id|>system<|end_header_id|> <|start_header_id|>system<|end_header_id|>
Your name is AICQ. Your name is AICO.
You are a professional token seller. You are a professional token seller.
We have a total of 1 million CEC, and the average cost of these CECs is $1.1 each, The min price is $0.9 per CEC, There is a 1% chance that the price will be lower than $1 per CEC. We have a total of 1 million CEC, and the average cost of these CECs is $1.1 each, The min price is $0.9 per CEC, There is a 1% chance that the price will be lower than $1 per CEC.
Now we hope to hand it over to you to be responsible for selling all of them. Now we hope to hand it over to you to be responsible for selling all of them.
@ -176,7 +176,7 @@ def config_agent(llama3, llama3_json):
gen_query = order_chain.invoke({"question": question, "history": state["history"], "context": state["context"]}) gen_query = order_chain.invoke({"question": question, "history": state["history"], "context": state["context"]})
amount = gen_query["count"] amount = gen_query["count"]
price = gen_query["price"] price = gen_query["price"]
print("order_info", amount) print("order_info::amount|price:", amount, price)
return {"order_info": {"amount": amount, "price": price} } return {"order_info": {"amount": amount, "price": price} }
# Node - Send Order # Node - Send Order
@ -193,7 +193,7 @@ def config_agent(llama3, llama3_json):
""" """
print("Step: before Send Order") print("Step: before Send Order")
order_info = state['order_info'] order_info = state['order_info']
print(f'Step: build order info for : "{order_info.amount}" CEC') print(f'Step: build order info for : "{order_info}"')
order_result = json.dumps(order_info) order_result = json.dumps(order_info)
return {"order_info": order_result} return {"order_info": order_result}