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

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-05-22 10:46:03 +08:00
parent 3d418d95c5
commit 3b9c28fae9
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/llm/llama_index_impl.py
View File

@@ -95,7 +95,7 @@ async def llama_index_complete_if_cache(
prompt: str, prompt: str,
system_prompt: Optional[str] = None, system_prompt: Optional[str] = None,
history_messages: List[dict] = [], history_messages: List[dict] = [],
chat_kwargs = {}, chat_kwargs={},
) -> str: ) -> str:
"""Complete the prompt using LlamaIndex.""" """Complete the prompt using LlamaIndex."""
try: try:
@@ -122,7 +122,9 @@ async def llama_index_complete_if_cache(
# Add current prompt # Add current prompt
formatted_messages.append(ChatMessage(role=MessageRole.USER, content=prompt)) formatted_messages.append(ChatMessage(role=MessageRole.USER, content=prompt))
response: ChatResponse = await model.achat(messages=formatted_messages, **chat_kwargs) response: ChatResponse = await model.achat(
messages=formatted_messages, **chat_kwargs
)
# In newer versions, the response is in message.content # In newer versions, the response is in message.content
content = response.message.content content = response.message.content