mjschock commited on
Commit
87a85e7
·
unverified ·
1 Parent(s): d129166

Enable debug mode for LiteLLM in main.py to enhance logging capabilities. This change allows for improved traceability during execution.

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -8,10 +8,13 @@ from typing import AsyncIterator, Optional, TypedDict
8
  from dotenv import find_dotenv, load_dotenv
9
  from langgraph.checkpoint.memory import MemorySaver
10
  from langgraph.graph import END, START, StateGraph
 
11
  from smolagents import CodeAgent, LiteLLMModel
12
  from smolagents.memory import ActionStep, FinalAnswerStep
13
  from smolagents.monitoring import LogLevel
14
 
 
 
15
  # Configure logging
16
  logging.basicConfig(
17
  level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
 
8
  from dotenv import find_dotenv, load_dotenv
9
  from langgraph.checkpoint.memory import MemorySaver
10
  from langgraph.graph import END, START, StateGraph
11
+ import litellm
12
  from smolagents import CodeAgent, LiteLLMModel
13
  from smolagents.memory import ActionStep, FinalAnswerStep
14
  from smolagents.monitoring import LogLevel
15
 
16
+ litellm._turn_on_debug()
17
+
18
  # Configure logging
19
  logging.basicConfig(
20
  level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"