pjpjq commited on
Commit
b386b30
·
1 Parent(s): fa13a08

Make API_KEY/API_BASE/SUMMARY_MODEL_NAME optional, fallback to DR_MODEL_* vars

Browse files
Files changed (1) hide show
  1. visit.py +4 -3
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
- api_key = os.environ.get("API_KEY")
101
- url_llm = os.environ.get("API_BASE")
102
- model_name = os.environ.get("SUMMARY_MODEL_NAME", "qwen/qwen3-30b-a3b-instruct-2507")
 
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,