rsrini7 commited on
Commit
c5a47d3
·
1 Parent(s): 2848e7f

refactor(llm_helper): extract OpenRouter base URL to constant

Browse files

Extract the OpenRouter base URL into a constant to improve code maintainability and avoid hardcoding the URL in multiple places.

Files changed (1) hide show
  1. 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=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,