shamik
commited on
fix: multiagent module imports.
Browse files
src/agent_hackathon/multiagent.py
CHANGED
|
@@ -1,16 +1,14 @@
|
|
| 1 |
import asyncio
|
| 2 |
-
import os
|
| 3 |
from datetime import date
|
| 4 |
|
| 5 |
-
from consts import PROJECT_ROOT_DIR
|
| 6 |
-
|
| 7 |
-
# from dotenv import find_dotenv, load_dotenv
|
| 8 |
-
from generate_arxiv_responses import ArxivResponseGenerator
|
| 9 |
from llama_index.core.agent.workflow import AgentWorkflow, ReActAgent
|
| 10 |
-
from llama_index.core.tools import FunctionTool
|
| 11 |
from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
|
| 12 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
from src.agent_hackathon.logger import get_logger
|
| 15 |
|
| 16 |
# _ = load_dotenv(dotenv_path=find_dotenv(raise_error_if_not_found=False), override=True)
|
|
@@ -30,7 +28,7 @@ class MultiAgentWorkflow:
|
|
| 30 |
# provider="nebius",
|
| 31 |
temperature=0.1,
|
| 32 |
top_p=0.95,
|
| 33 |
-
max_tokens=8192
|
| 34 |
# api_key=os.getenv(key="NEBIUS_API_KEY"),
|
| 35 |
# base_url="https://api.studio.nebius.com/v1/",
|
| 36 |
)
|
|
|
|
| 1 |
import asyncio
|
|
|
|
| 2 |
from datetime import date
|
| 3 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
from llama_index.core.agent.workflow import AgentWorkflow, ReActAgent
|
|
|
|
| 5 |
from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
|
| 6 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
| 7 |
|
| 8 |
+
from src.agent_hackathon.consts import PROJECT_ROOT_DIR
|
| 9 |
+
|
| 10 |
+
# from dotenv import find_dotenv, load_dotenv
|
| 11 |
+
from src.agent_hackathon.generate_arxiv_responses import ArxivResponseGenerator
|
| 12 |
from src.agent_hackathon.logger import get_logger
|
| 13 |
|
| 14 |
# _ = load_dotenv(dotenv_path=find_dotenv(raise_error_if_not_found=False), override=True)
|
|
|
|
| 28 |
# provider="nebius",
|
| 29 |
temperature=0.1,
|
| 30 |
top_p=0.95,
|
| 31 |
+
max_tokens=8192,
|
| 32 |
# api_key=os.getenv(key="NEBIUS_API_KEY"),
|
| 33 |
# base_url="https://api.studio.nebius.com/v1/",
|
| 34 |
)
|