Spaces:
Runtime error
Runtime error
Leonardo
commited on
Update app.py
Browse files
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/
|
| 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
|