Spaces:
Running
Running
refactor(llm_helper): extract OpenRouter base URL to constant
Browse filesExtract the OpenRouter base URL into a constant to improve code maintainability and avoid hardcoding the URL in multiple places.
- helpers/llm_helper.py +2 -2
helpers/llm_helper.py
CHANGED
@@ -23,6 +23,7 @@ OLLAMA_MODEL_REGEX = re.compile(r'[a-zA-Z0-9._:-]+$')
|
|
23 |
# 94 characters long, only containing alphanumeric characters, hyphens, and underscores
|
24 |
API_KEY_REGEX = re.compile(r'^[a-zA-Z0-9_-]{6,94}$')
|
25 |
REQUEST_TIMEOUT = 35
|
|
|
26 |
|
27 |
|
28 |
logger = logging.getLogger(__name__)
|
@@ -194,10 +195,9 @@ def get_langchain_llm(
|
|
194 |
|
195 |
logger.debug('Getting LLM via OpenRouter: %s', model)
|
196 |
openrouter_api_key = api_key
|
197 |
-
base_url = 'https://openrouter.ai/api/v1'
|
198 |
|
199 |
return ChatOpenAI(
|
200 |
-
base_url=
|
201 |
openai_api_key=openrouter_api_key,
|
202 |
model_name=model,
|
203 |
temperature=GlobalConfig.LLM_MODEL_TEMPERATURE,
|
|
|
23 |
# 94 characters long, only containing alphanumeric characters, hyphens, and underscores
|
24 |
API_KEY_REGEX = re.compile(r'^[a-zA-Z0-9_-]{6,94}$')
|
25 |
REQUEST_TIMEOUT = 35
|
26 |
+
OPENROUTER_BASE_URL = 'https://openrouter.ai/api/v1'
|
27 |
|
28 |
|
29 |
logger = logging.getLogger(__name__)
|
|
|
195 |
|
196 |
logger.debug('Getting LLM via OpenRouter: %s', model)
|
197 |
openrouter_api_key = api_key
|
|
|
198 |
|
199 |
return ChatOpenAI(
|
200 |
+
base_url=OPENROUTER_BASE_URL,
|
201 |
openai_api_key=openrouter_api_key,
|
202 |
model_name=model,
|
203 |
temperature=GlobalConfig.LLM_MODEL_TEMPERATURE,
|