ziadsameh32 commited on
Commit
32a256f
·
1 Parent(s): b7c5ad1

Initial FastAPI CrewAI setup

Browse files
agents/books/apa_agent.py CHANGED
@@ -5,7 +5,7 @@ from schemas.books.sources_schema import (
5
  SourceForAgent,
6
  SourceMetadataAgentResult,
7
  )
8
- from modules import llm_open_grok
9
 
10
 
11
  # ============================================================
@@ -26,7 +26,7 @@ metadata_agent = Agent(
26
  "If something is unclear, ambiguous, or missing, you explicitly mark it as missing. "
27
  "You prefer returning invalid metadata over returning incorrect metadata."
28
  ),
29
- llm=llm_open_grok,
30
  verbose=False,
31
  memory=False,
32
  allow_delegation=False,
 
5
  SourceForAgent,
6
  SourceMetadataAgentResult,
7
  )
8
+ from modules import llm_oss_120b
9
 
10
 
11
  # ============================================================
 
26
  "If something is unclear, ambiguous, or missing, you explicitly mark it as missing. "
27
  "You prefer returning invalid metadata over returning incorrect metadata."
28
  ),
29
+ llm=llm_oss_120b,
30
  verbose=False,
31
  memory=False,
32
  allow_delegation=False,
modules/__init__.py CHANGED
@@ -5,6 +5,7 @@ from .env import (
5
  llm_4m,
6
  llm_5n,
7
  llm_oss,
 
8
  llm_open_c3,
9
  llm_open_c35,
10
  llm_open_g25pro,
 
5
  llm_4m,
6
  llm_5n,
7
  llm_oss,
8
+ llm_oss_120b,
9
  llm_open_c3,
10
  llm_open_c35,
11
  llm_open_g25pro,
modules/env.py CHANGED
@@ -24,6 +24,11 @@ llm_oss = LLM(
24
  api_key=openrouter_key,
25
  base_url="https://openrouter.ai/api/v1",
26
  )
 
 
 
 
 
27
 
28
 
29
  llm_g = LLM(
 
24
  api_key=openrouter_key,
25
  base_url="https://openrouter.ai/api/v1",
26
  )
27
+ llm_oss_120b = LLM(
28
+ model="openai/gpt-oss-120b:nitro",
29
+ api_key=openrouter_key,
30
+ base_url="https://openrouter.ai/api/v1",
31
+ )
32
 
33
 
34
  llm_g = LLM(