SimpleSam commited on
Commit
4f184a0
·
verified ·
1 Parent(s): 801b7b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
7
 
8
  from Gradio_UI import GradioUI
9
 
@@ -38,7 +39,9 @@ def get_property(location: str, title: str, count: int) -> str:
38
  """
39
  A tool to that shows property.
40
  Args:
41
- location: The location where property is supposed to be searched
 
 
42
  """
43
  try:
44
  #
@@ -49,6 +52,7 @@ def get_property(location: str, title: str, count: int) -> str:
49
  return f"error fetching results for location '{location}': {str(e)}"
50
 
51
  final_answer = FinalAnswerTool()
 
52
 
53
  # 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:
54
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
@@ -69,7 +73,7 @@ with open("prompts.yaml", 'r') as stream:
69
 
70
  agent = CodeAgent(
71
  model=model,
72
- tools=[final_answer], ## add your tools here (don't remove final answer)
73
  max_steps=6,
74
  verbosity_level=1,
75
  grammar=None,
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ from tools.manyata_search import ManyataSearchTool
8
 
9
  from Gradio_UI import GradioUI
10
 
 
39
  """
40
  A tool to that shows property.
41
  Args:
42
+ location: The location where property is supposed to be searched
43
+ title: a word within the title of the property
44
+ count: the number of results expected
45
  """
46
  try:
47
  #
 
52
  return f"error fetching results for location '{location}': {str(e)}"
53
 
54
  final_answer = FinalAnswerTool()
55
+ manyata_search = ManyataSearchTool()
56
 
57
  # 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:
58
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 
73
 
74
  agent = CodeAgent(
75
  model=model,
76
+ tools=[final_answer, manyata_search], ## add your tools here (don't remove final answer)
77
  max_steps=6,
78
  verbosity_level=1,
79
  grammar=None,