reedmayhew commited on
Commit
4a30274
·
verified ·
1 Parent(s): 13528d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import requests
7
  openai_api_key = os.getenv("OPENROUTER_API_KEY")
8
  openai_base_url = os.getenv("OPENAI_BASE_URL")
9
  ai_model = os.getenv("AI_MODEL")
 
10
 
11
  # Configure the OpenAI client with your custom API endpoint and API key.
12
  client = OpenAI(base_url=openai_base_url, api_key=openai_api_key)
@@ -93,7 +94,7 @@ def think(request):
93
 
94
 
95
  # R1 will reliably return "done" for the content portion of the response
96
- reasoning_response = do_req("deepseek/deepseek-r1", request, True)
97
  reasoning = reasoning_response.json()['choices'][0]['message']['reasoning']
98
  return reasoning
99
 
 
7
  openai_api_key = os.getenv("OPENROUTER_API_KEY")
8
  openai_base_url = os.getenv("OPENAI_BASE_URL")
9
  ai_model = os.getenv("AI_MODEL")
10
+ reasoning_ai_model = os.getenv("REASONING_AI_MODEL")
11
 
12
  # Configure the OpenAI client with your custom API endpoint and API key.
13
  client = OpenAI(base_url=openai_base_url, api_key=openai_api_key)
 
94
 
95
 
96
  # R1 will reliably return "done" for the content portion of the response
97
+ reasoning_response = do_req(reasoning_ai_model, request, True)
98
  reasoning = reasoning_response.json()['choices'][0]['message']['reasoning']
99
  return reasoning
100