Update agent.py
Browse files
agent.py
CHANGED
|
@@ -6,7 +6,9 @@ import logfire
|
|
| 6 |
|
| 7 |
load_dotenv()
|
| 8 |
|
| 9 |
-
logfire.configure(token=os.getenv("LOGFIRE_API_KEY")
|
|
|
|
|
|
|
| 10 |
|
| 11 |
model = LiteLLMModel(
|
| 12 |
model_id="openrouter/openai/gpt-4.1-mini",
|
|
|
|
| 6 |
|
| 7 |
load_dotenv()
|
| 8 |
|
| 9 |
+
logfire.configure(token=os.getenv("LOGFIRE_API_KEY"))
|
| 10 |
+
litellm.callbacks = ["logfire"]
|
| 11 |
+
litellm.success_callback = ["logfire"]
|
| 12 |
|
| 13 |
model = LiteLLMModel(
|
| 14 |
model_id="openrouter/openai/gpt-4.1-mini",
|