galactromedaNPMAIECOSYSTEM commited on
Commit
cd80d7a
·
verified ·
1 Parent(s): 3b2ca38

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -3
main.py CHANGED
@@ -30,13 +30,15 @@ async def generate_response(request: PromptRequest):
30
  temperature=request.temperature,
31
  base_url="http://localhost:11434"
32
  )
33
- tool_prompt = f"System Role: You are an autonomous AI Agent with real-time internet access. Current Date: {{today_date}} TOOL_DEFINITION: - Name: Search_tool - Activation Command: Search [Your Query Here]"
34
- response = llm.invoke(f'{{tool_prompt}}, User-Query:-{{request.prompt}}')
 
35
 
36
  if "Search " in response:
37
  new_query = response.removeprefix("Search ")
38
  search_response = search_tool(query=new_query)
39
- new_response = llm.invoke(f"Extra_information:- {{search_response}} User-Query:- {{request.prompt}}")
 
40
  return {"response": new_response}
41
  else:
42
  return {"response": response}
 
30
  temperature=request.temperature,
31
  base_url="http://localhost:11434"
32
  )
33
+ tool_prompt = f"System Role: You are an autonomous AI Agent with real-time internet access. Current Date: {today_date} TOOL_DEFINITION: - Name: Search_tool - Activation Command: Search [Your Query Here]"
34
+ response = llm.invoke(f'{tool_prompt}, User-Query:-{request.prompt}')
35
+ print(response)
36
 
37
  if "Search " in response:
38
  new_query = response.removeprefix("Search ")
39
  search_response = search_tool(query=new_query)
40
+ new_response = llm.invoke(f"Extra_information:- {search_response} User-Query:- {request.prompt}")
41
+ print(new_response)
42
  return {"response": new_response}
43
  else:
44
  return {"response": response}