gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-07-24 12:25:50 +08:00
parent d979e9078f
commit 2767212ba0
3 changed files with 28 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/api/lightrag_server.py
View file

@ -89,7 +89,13 @@ def create_app(args):
]:
raise Exception("llm binding not supported")
if args.embedding_binding not in ["lollms", "ollama", "openai", "azure_openai", "jina"]:
if args.embedding_binding not in [
"lollms",
"ollama",
"openai",
"azure_openai",
"jina",
]:
raise Exception("embedding binding not supported")
# Set default hosts if not provided

14
lightrag/llm/jina.py
View file

@ -28,7 +28,7 @@ async def fetch_data(url, headers, data):
request_info=response.request_info,
history=response.history,
status=response.status,
message=f"Jina API error: {error_text}"
message=f"Jina API error: {error_text}",
)
response_json = await response.json()
data_list = response_json.get("data", [])
@ -89,7 +89,9 @@ async def jina_embed(
if late_chunking:
data["late_chunking"] = late_chunking
logger.debug(f"Jina embedding request: {len(texts)} texts, dimensions: {dimensions}")
logger.debug(
f"Jina embedding request: {len(texts)} texts, dimensions: {dimensions}"
)
try:
data_list = await fetch_data(url, headers, data)
@ -99,8 +101,12 @@ async def jina_embed(
raise ValueError("Jina API returned empty data list")
if len(data_list) != len(texts):
logger.error(f"Jina API returned {len(data_list)} embeddings for {len(texts)} texts")
raise ValueError(f"Jina API returned {len(data_list)} embeddings for {len(texts)} texts")
logger.error(
f"Jina API returned {len(data_list)} embeddings for {len(texts)} texts"
)
raise ValueError(
f"Jina API returned {len(data_list)} embeddings for {len(texts)} texts"
)
embeddings = np.array([dp["embedding"] for dp in data_list])
logger.debug(f"Jina embeddings generated: shape {embeddings.shape}")