gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update webui assets

Browse Source
This commit is contained in:
yangdx
2025-03-22 00:36:38 +08:00
parent b21c32645d
commit dc99b714ba
3 changed files with 176 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

192
lightrag/api/webui/assets/index-C-CHRwmZ.js → lightrag/api/webui/assets/index-BItOVH8B.js generated
View File

File diff suppressed because one or more lines are too long

2
lightrag/api/webui/index.html generated
View File

@@ -8,7 +8,7 @@
<link rel="icon" type="image/svg+xml" href="logo.png" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>Lightrag</title>
<script type="module" crossorigin src="/webui/assets/index-C-CHRwmZ.js"></script>
<script type="module" crossorigin src="/webui/assets/index-BItOVH8B.js"></script>
<link rel="stylesheet" crossorigin href="/webui/assets/index-BViPRMGA.css">
</head>
<body>

68
lightrag/llm/anthropic.py
View File

@@ -38,11 +38,14 @@ from lightrag.utils import (
)
from lightrag.api import __api_version__
# Custom exception for retry mechanism
class InvalidResponseError(Exception):
"""Custom exception class for triggering retry mechanism"""
pass
# Core Anthropic completion function with retry
@retry(
stop=stop_after_attempt(3),
@@ -96,10 +99,7 @@ async def anthropic_complete_if_cache(
try:
response = await anthropic_async_client.messages.create(
model=model,
messages=messages,
stream=True,
**kwargs
model=model, messages=messages, stream=True, **kwargs
)
except APIConnectionError as e:
logger.error(f"Anthropic API Connection Error: {e}")
@@ -119,7 +119,11 @@ async def anthropic_complete_if_cache(
async def stream_response():
try:
async for event in response:
content = event.delta.text if hasattr(event, "delta") and event.delta.text else None
content = (
event.delta.text
if hasattr(event, "delta") and event.delta.text
else None
)
if content is None:
continue
if r"\u" in content:
@@ -131,6 +135,7 @@ async def anthropic_complete_if_cache(
return stream_response()
# Generic Anthropic completion function
async def anthropic_complete(
prompt: str,
@@ -149,6 +154,7 @@ async def anthropic_complete(
**kwargs,
)
# Claude 3 Opus specific completion
async def claude_3_opus_complete(
prompt: str,
@@ -166,6 +172,7 @@ async def claude_3_opus_complete(
**kwargs,
)
# Claude 3 Sonnet specific completion
async def claude_3_sonnet_complete(
prompt: str,
@@ -183,6 +190,7 @@ async def claude_3_sonnet_complete(
**kwargs,
)
# Claude 3 Haiku specific completion
async def claude_3_haiku_complete(
prompt: str,
@@ -200,6 +208,7 @@ async def claude_3_haiku_complete(
**kwargs,
)
# Embedding function (placeholder, as Anthropic does not provide embeddings)
@retry(
stop=stop_after_attempt(3),
@@ -230,7 +239,9 @@ async def anthropic_embed(
api_key = os.environ.get("VOYAGE_API_KEY")
if not api_key:
logger.error("VOYAGE_API_KEY environment variable not set")
raise ValueError("VOYAGE_API_KEY environment variable is required for embeddings")
raise ValueError(
"VOYAGE_API_KEY environment variable is required for embeddings"
)
try:
# Initialize Voyage AI client
@@ -240,7 +251,7 @@ async def anthropic_embed(
result = voyage_client.embed(
texts,
model=model,
input_type="document" # Assuming document context; could be made configurable
input_type="document", # Assuming document context; could be made configurable
)
# Convert list of embeddings to numpy array
@@ -255,17 +266,46 @@ async def anthropic_embed(
logger.error(f"Voyage AI embedding failed: {str(e)}")
raise
# Optional: a helper function to get available embedding models
def get_available_embedding_models() -> dict[str, dict]:
"""
Returns a dictionary of available Voyage AI embedding models and their properties.
"""
return {
"voyage-3-large": {"context_length": 32000, "dimension": 1024, "description": "Best general-purpose and multilingual"},
"voyage-3": {"context_length": 32000, "dimension": 1024, "description": "General-purpose and multilingual"},
"voyage-3-lite": {"context_length": 32000, "dimension": 512, "description": "Optimized for latency and cost"},
"voyage-code-3": {"context_length": 32000, "dimension": 1024, "description": "Optimized for code"},
"voyage-finance-2": {"context_length": 32000, "dimension": 1024, "description": "Optimized for finance"},
"voyage-law-2": {"context_length": 16000, "dimension": 1024, "description": "Optimized for legal"},
"voyage-multimodal-3": {"context_length": 32000, "dimension": 1024, "description": "Multimodal text and images"},
"voyage-3-large": {
"context_length": 32000,
"dimension": 1024,
"description": "Best general-purpose and multilingual",
},
"voyage-3": {
"context_length": 32000,
"dimension": 1024,
"description": "General-purpose and multilingual",
},
"voyage-3-lite": {
"context_length": 32000,
"dimension": 512,
"description": "Optimized for latency and cost",
},
"voyage-code-3": {
"context_length": 32000,
"dimension": 1024,
"description": "Optimized for code",
},
"voyage-finance-2": {
"context_length": 32000,
"dimension": 1024,
"description": "Optimized for finance",
},
"voyage-law-2": {
"context_length": 16000,
"dimension": 1024,
"description": "Optimized for legal",
},
"voyage-multimodal-3": {
"context_length": 32000,
"dimension": 1024,
"description": "Multimodal text and images",
},
}