sgur11 commited on
Commit
08d1415
·
verified ·
1 Parent(s): 49cf91c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,7 +18,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
18
  """
19
  search = ("Find top 10 colleges offereing %s in %s" % arg1, arg2)
20
  content = DuckDuckGoSearchTool(search)
21
- return "What magic will you build ?"
22
 
23
  @tool
24
  def get_current_time_in_timezone(timezone: str) -> str:
@@ -39,6 +39,7 @@ timezone = 'Asia/Kolkata'
39
 
40
  final_answer = FinalAnswerTool()
41
  timezone_tool = get_current_time_in_timezone(timezone)
 
42
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
43
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
44
 
@@ -58,7 +59,7 @@ with open("prompts.yaml", 'r') as stream:
58
 
59
  agent = CodeAgent(
60
  model=model,
61
- tools=[final_answer,], ## add your tools here (don't remove final answer)
62
  max_steps=6,
63
  verbosity_level=1,
64
  grammar=None,
 
18
  """
19
  search = ("Find top 10 colleges offereing %s in %s" % arg1, arg2)
20
  content = DuckDuckGoSearchTool(search)
21
+ return content
22
 
23
  @tool
24
  def get_current_time_in_timezone(timezone: str) -> str:
 
39
 
40
  final_answer = FinalAnswerTool()
41
  timezone_tool = get_current_time_in_timezone(timezone)
42
+ search = my_custom_tool('B.tech','Punjab')
43
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
44
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
45
 
 
59
 
60
  agent = CodeAgent(
61
  model=model,
62
+ tools=[final_answer,search], ## add your tools here (don't remove final answer)
63
  max_steps=6,
64
  verbosity_level=1,
65
  grammar=None,