Enable "just an Anthropic proxy" mode (possibility not to remap to neither OpenAI nor Gemini)
This commit is contained in:
parent
e9c8cf8de6
commit
8876b4dbd3
1 changed files with 6 additions and 1 deletions
|
|
@ -205,8 +205,13 @@ class MessagesRequest(BaseModel):
|
||||||
|
|
||||||
# --- Mapping Logic --- START ---
|
# --- Mapping Logic --- START ---
|
||||||
mapped = False
|
mapped = False
|
||||||
|
if PREFERRED_PROVIDER == "anthropic":
|
||||||
|
# Don't remap to big/small models, just add the prefix
|
||||||
|
new_model = f"anthropic/{clean_v}"
|
||||||
|
mapped = True
|
||||||
|
|
||||||
# Map Haiku to SMALL_MODEL based on provider preference
|
# Map Haiku to SMALL_MODEL based on provider preference
|
||||||
if 'haiku' in clean_v.lower():
|
elif 'haiku' in clean_v.lower():
|
||||||
if PREFERRED_PROVIDER == "google" and SMALL_MODEL in GEMINI_MODELS:
|
if PREFERRED_PROVIDER == "google" and SMALL_MODEL in GEMINI_MODELS:
|
||||||
new_model = f"gemini/{SMALL_MODEL}"
|
new_model = f"gemini/{SMALL_MODEL}"
|
||||||
mapped = True
|
mapped = True
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue