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 |
)
|