yooke commited on
Commit
2885823
·
verified ·
1 Parent(s): 8f41e23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,6 +12,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
12
  # --- Basic Agent Definition ---
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
14
  DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 
15
  class BasicAgent:
16
  def __init__(self):
17
  print("BasicAgent initialized.")
@@ -22,7 +23,8 @@ class BasicAgent:
22
 
23
  model = OpenAIServerModel(
24
  model_id="deepseek-chat",
25
- api_key=DEEPSEEK_API_KEY
 
26
  )
27
  # Initialize the search tool
28
  search_tool = DuckDuckGoSearchTool()
 
12
  # --- Basic Agent Definition ---
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
14
  DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
15
+ DEEPSEEK_API_BASE = os.getenv("DEEPSEEK_API_BASE", "https://api.deepseek.com/v1")
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
 
23
 
24
  model = OpenAIServerModel(
25
  model_id="deepseek-chat",
26
+ api_key=DEEPSEEK_API_KEY,
27
+ base_url=DEEPSEEK_API_BASE
28
  )
29
  # Initialize the search tool
30
  search_tool = DuckDuckGoSearchTool()