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")
|
model = payload.get("model")
|
||||||
doc = payload.get("input")
|
doc = payload.get("input")
|
||||||
|
|
||||||
if ":latest" in model:
|
|
||||||
model = model[:-7]
|
|
||||||
|
|
||||||
if not model:
|
if not model:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
|
|
@ -1414,7 +1411,7 @@ async def openai_embedding_proxy(request: Request):
|
||||||
# 2. Endpoint logic
|
# 2. Endpoint logic
|
||||||
endpoint = await choose_endpoint(model)
|
endpoint = await choose_endpoint(model)
|
||||||
await increment_usage(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]
|
api_key = config.api_keys[endpoint]
|
||||||
else:
|
else:
|
||||||
api_key = "ollama"
|
api_key = "ollama"
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue