mirror of
https://github.com/open-webui/open-webui
synced 2025-06-10 00:17:52 +00:00
refac
This commit is contained in:
parent
036ce12dd9
commit
9306ae5972
@ -818,63 +818,33 @@ def generate_embeddings(
|
|||||||
text = f"{prefix}{text}"
|
text = f"{prefix}{text}"
|
||||||
|
|
||||||
if engine == "ollama":
|
if engine == "ollama":
|
||||||
if isinstance(text, list):
|
embeddings = generate_ollama_batch_embeddings(
|
||||||
embeddings = generate_ollama_batch_embeddings(
|
**{
|
||||||
**{
|
"model": model,
|
||||||
"model": model,
|
"texts": text if isinstance(text, list) else [text],
|
||||||
"texts": text,
|
"url": url,
|
||||||
"url": url,
|
"key": key,
|
||||||
"key": key,
|
"prefix": prefix,
|
||||||
"prefix": prefix,
|
"user": user,
|
||||||
"user": user,
|
}
|
||||||
}
|
)
|
||||||
)
|
|
||||||
else:
|
|
||||||
embeddings = generate_ollama_batch_embeddings(
|
|
||||||
**{
|
|
||||||
"model": model,
|
|
||||||
"texts": [text],
|
|
||||||
"url": url,
|
|
||||||
"key": key,
|
|
||||||
"prefix": prefix,
|
|
||||||
"user": user,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
return embeddings[0] if isinstance(text, str) else embeddings
|
return embeddings[0] if isinstance(text, str) else embeddings
|
||||||
elif engine == "openai":
|
elif engine == "openai":
|
||||||
if isinstance(text, list):
|
embeddings = generate_openai_batch_embeddings(
|
||||||
embeddings = generate_openai_batch_embeddings(
|
model, text if isinstance(text, list) else [text], url, key, prefix, user
|
||||||
model, text, url, key, prefix, user
|
)
|
||||||
)
|
|
||||||
else:
|
|
||||||
embeddings = generate_openai_batch_embeddings(
|
|
||||||
model, [text], url, key, prefix, user
|
|
||||||
)
|
|
||||||
return embeddings[0] if isinstance(text, str) else embeddings
|
return embeddings[0] if isinstance(text, str) else embeddings
|
||||||
elif engine == "azure_openai":
|
elif engine == "azure_openai":
|
||||||
azure_api_version = kwargs.get("azure_api_version", "")
|
azure_api_version = kwargs.get("azure_api_version", "")
|
||||||
if isinstance(text, list):
|
embeddings = generate_azure_openai_batch_embeddings(
|
||||||
embeddings = generate_azure_openai_batch_embeddings(
|
model,
|
||||||
model,
|
text if isinstance(text, list) else [text],
|
||||||
text,
|
url,
|
||||||
url,
|
key,
|
||||||
key,
|
azure_api_version,
|
||||||
model,
|
prefix,
|
||||||
azure_api_version,
|
user,
|
||||||
prefix,
|
)
|
||||||
user,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
embeddings = generate_azure_openai_batch_embeddings(
|
|
||||||
model,
|
|
||||||
[text],
|
|
||||||
url,
|
|
||||||
key,
|
|
||||||
model,
|
|
||||||
azure_api_version,
|
|
||||||
prefix,
|
|
||||||
user,
|
|
||||||
)
|
|
||||||
return embeddings[0] if isinstance(text, str) else embeddings
|
return embeddings[0] if isinstance(text, str) else embeddings
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user