rollback - needs more logic in v1/embedding
This commit is contained in:
parent
9f90bc9cd0
commit
b9933e000f
1 changed files with 1 additions and 4 deletions
|
|
@ -1396,9 +1396,6 @@ async def openai_embedding_proxy(request: Request):
|
|||
|
||||
model = payload.get("model")
|
||||
doc = payload.get("input")
|
||||
|
||||
if ":latest" in model:
|
||||
model = model[:-7]
|
||||
|
||||
if not model:
|
||||
raise HTTPException(
|
||||
|
|
@ -1414,7 +1411,7 @@ async def openai_embedding_proxy(request: Request):
|
|||
# 2. Endpoint logic
|
||||
endpoint = await choose_endpoint(model)
|
||||
await increment_usage(endpoint, model)
|
||||
if "/v1" in endpoint:
|
||||
if "/v1" in endpoint and is_ext_openai_endpoint(endpoint):
|
||||
api_key = config.api_keys[endpoint]
|
||||
else:
|
||||
api_key = "ollama"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue