ofermend commited on
Commit
a21574c
·
1 Parent(s): d53865c

updated prompt

Browse files
Files changed (1) hide show
  1. agent.py +1 -4
agent.py CHANGED
@@ -77,10 +77,8 @@ def create_assistant_tools(cfg):
77
 
78
  return f"FMP API returned error {response.status_code}. This tool does not work."
79
 
80
-
81
-
82
  class QueryTranscriptsArgs(BaseModel):
83
- query: str = Field(..., description="The user query.")
84
  year: int = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)}.")
85
  ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {tickers.keys()}.")
86
 
@@ -130,7 +128,6 @@ def initialize_agent(_cfg, agent_progress_callback=None):
130
  - When querying a tool for a numeric value or KPI, use a concise and non-ambiguous description of what you are looking for.
131
  - If you calculate a metric, make sure you have all the necessary information to complete the calculation. Don't guess.
132
  """
133
- # - Use the get_income_statement tool only to receive financial data like revenue, net income, and other financial metrics for a specific company and year.
134
 
135
  agent = Agent(
136
  tools=create_assistant_tools(_cfg),
 
77
 
78
  return f"FMP API returned error {response.status_code}. This tool does not work."
79
 
 
 
80
  class QueryTranscriptsArgs(BaseModel):
81
+ query: str = Field(..., description="The user query, always in the form of a question", examples=["what are the risks reported?", "who are the competitors?"])
82
  year: int = Field(..., description=f"The year this query relates to. An integer between {min(years)} and {max(years)}.")
83
  ticker: str = Field(..., description=f"The company ticker this query relates to. Must be a valid ticket symbol from the list {tickers.keys()}.")
84
 
 
128
  - When querying a tool for a numeric value or KPI, use a concise and non-ambiguous description of what you are looking for.
129
  - If you calculate a metric, make sure you have all the necessary information to complete the calculation. Don't guess.
130
  """
 
131
 
132
  agent = Agent(
133
  tools=create_assistant_tools(_cfg),