Spaces:
Paused
Paused
Make API_KEY/API_BASE/SUMMARY_MODEL_NAME optional, fallback to DR_MODEL_* vars
Browse files
visit.py
CHANGED
|
@@ -97,9 +97,10 @@ class Visit(BaseTool):
|
|
| 97 |
return response.strip()
|
| 98 |
|
| 99 |
def call_server(self, msgs, max_retries=2):
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
|
|
|
| 103 |
|
| 104 |
client = OpenAI(
|
| 105 |
api_key=api_key,
|
|
|
|
| 97 |
return response.strip()
|
| 98 |
|
| 99 |
def call_server(self, msgs, max_retries=2):
|
| 100 |
+
# Use summary model config, fallback to main model config
|
| 101 |
+
api_key = os.environ.get("API_KEY") or os.environ.get("DR_MODEL_API_KEY")
|
| 102 |
+
url_llm = os.environ.get("API_BASE") or os.environ.get("DR_MODEL_SERVER")
|
| 103 |
+
model_name = os.environ.get("SUMMARY_MODEL_NAME") or os.environ.get("DR_MODEL_NAME", "qwen/qwen3-30b-a3b-instruct-2507")
|
| 104 |
|
| 105 |
client = OpenAI(
|
| 106 |
api_key=api_key,
|