Fix linting
This commit is contained in:
parent
ce28f30ca6
commit
d95efcb9ad
1 changed files with 9 additions and 2 deletions
|
|
@ -644,13 +644,20 @@ def create_app(args):
|
||||||
from lightrag.llm.jina import jina_embed
|
from lightrag.llm.jina import jina_embed
|
||||||
|
|
||||||
return await jina_embed(
|
return await jina_embed(
|
||||||
texts, embedding_dim=embedding_dim, base_url=host, api_key=api_key
|
texts,
|
||||||
|
embedding_dim=embedding_dim,
|
||||||
|
base_url=host,
|
||||||
|
api_key=api_key,
|
||||||
)
|
)
|
||||||
else: # openai and compatible
|
else: # openai and compatible
|
||||||
from lightrag.llm.openai import openai_embed
|
from lightrag.llm.openai import openai_embed
|
||||||
|
|
||||||
return await openai_embed(
|
return await openai_embed(
|
||||||
texts, model=model, base_url=host, api_key=api_key, embedding_dim=embedding_dim
|
texts,
|
||||||
|
model=model,
|
||||||
|
base_url=host,
|
||||||
|
api_key=api_key,
|
||||||
|
embedding_dim=embedding_dim,
|
||||||
)
|
)
|
||||||
except ImportError as e:
|
except ImportError as e:
|
||||||
raise Exception(f"Failed to import {binding} embedding: {e}")
|
raise Exception(f"Failed to import {binding} embedding: {e}")
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue