Leonardo commited on
Commit
bf31f63
·
verified ·
1 Parent(s): efb258d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -24,6 +24,7 @@ from scripts.text_web_browser import (
24
  from scripts.visual_qa import visualizer
25
  # from scripts.frontmatter_tool import FrontmatterGeneratorTool
26
  from scripts.text_cleaner_tool import TextCleanerTool
 
27
 
28
  from smolagents import (
29
  CodeAgent,
@@ -197,6 +198,7 @@ class ToolRegistry:
197
  """
198
  return [
199
  TextCleanerTool(),
 
200
  ]
201
 
202
  @staticmethod
@@ -227,7 +229,7 @@ def create_agent():
227
  # Initialize model
228
  model = LiteLLMModel(
229
  custom_role_conversions=CUSTOM_ROLE_CONVERSIONS,
230
- model_id="openrouter/google/gemini-2.0-flash-001",
231
  )
232
 
233
  # Initialize tools
 
24
  from scripts.visual_qa import visualizer
25
  # from scripts.frontmatter_tool import FrontmatterGeneratorTool
26
  from scripts.text_cleaner_tool import TextCleanerTool
27
+ from scripts.document_tool import LegalDocumentTool
28
 
29
  from smolagents import (
30
  CodeAgent,
 
198
  """
199
  return [
200
  TextCleanerTool(),
201
+ LegalDocumentTool(),
202
  ]
203
 
204
  @staticmethod
 
229
  # Initialize model
230
  model = LiteLLMModel(
231
  custom_role_conversions=CUSTOM_ROLE_CONVERSIONS,
232
+ model_id="openrouter/deepseek/deepseek-v3-base:free",
233
  )
234
 
235
  # Initialize tools