Spaces:
Running
Running
Commit ·
32a256f
1
Parent(s): b7c5ad1
Initial FastAPI CrewAI setup
Browse files- agents/books/apa_agent.py +2 -2
- modules/__init__.py +1 -0
- modules/env.py +5 -0
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
|
| 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=
|
| 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(
|