ofermend commited on
Commit
c4b0cfc
1 Parent(s): fc8047d

updated tools

Browse files
Files changed (1) hide show
  1. agent.py +2 -3
agent.py CHANGED
@@ -164,6 +164,7 @@ def create_assistant_tools(cfg):
164
  )
165
 
166
  return (
 
167
  [tools_factory.create_tool(tool) for tool in [
168
  get_opinion_text,
169
  get_case_document_pdf,
@@ -173,9 +174,7 @@ def create_assistant_tools(cfg):
173
  validate_url
174
  ]] +
175
  tools_factory.standard_tools() +
176
- tools_factory.legal_tools() +
177
- tools_factory.guardrail_tools() +
178
- [ask_caselaw]
179
  )
180
 
181
  def get_agent_config() -> OmegaConf:
 
164
  )
165
 
166
  return (
167
+ [ask_caselaw] +
168
  [tools_factory.create_tool(tool) for tool in [
169
  get_opinion_text,
170
  get_case_document_pdf,
 
174
  validate_url
175
  ]] +
176
  tools_factory.standard_tools() +
177
+ tools_factory.legal_tools()
 
 
178
  )
179
 
180
  def get_agent_config() -> OmegaConf: