ar9avg commited on
Commit
68ebe84
·
1 Parent(s): cb9cfe8

fix: default LLM to HF Router + Qwen2.5-72B, no custom Space variables needed

Browse files
Files changed (1) hide show
  1. backend/env/sql_env.py +2 -2
backend/env/sql_env.py CHANGED
@@ -70,11 +70,11 @@ class RewardInfo(BaseModel):
70
  def _make_client() -> AsyncOpenAI:
71
  return AsyncOpenAI(
72
  api_key=os.environ.get("HF_TOKEN", ""),
73
- base_url=os.environ.get("API_BASE_URL", "https://api.openai.com/v1"),
74
  )
75
 
76
 
77
- _MODEL = os.environ.get("MODEL_NAME", "gpt-4o-mini")
78
 
79
  BASE_SYSTEM_PROMPT = """You are a SQL expert. Given a natural language question and a SQLite database schema, write a correct SQL query.
80
 
 
70
  def _make_client() -> AsyncOpenAI:
71
  return AsyncOpenAI(
72
  api_key=os.environ.get("HF_TOKEN", ""),
73
+ base_url=os.environ.get("API_BASE_URL", "https://router.huggingface.co/v1"),
74
  )
75
 
76
 
77
+ _MODEL = os.environ.get("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
78
 
79
  BASE_SYSTEM_PROMPT = """You are a SQL expert. Given a natural language question and a SQLite database schema, write a correct SQL query.
80