shamik commited on
Commit
c737a31
·
unverified ·
1 Parent(s): 8dbf816

fix: multiagent module imports.

Browse files
Files changed (1) hide show
  1. src/agent_hackathon/multiagent.py +5 -7
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
  )