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

Merge branch 'add-keyword-extraction-param-for-llm' into fix-mutable-default-param

Browse Source
This commit is contained in:
yangdx
2025-02-06 16:22:30 +08:00
parent eb5f57e989 eb9883d8da
commit 070501fdaa
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/api/lightrag_server.py
View File

@@ -17,6 +17,7 @@ import argparse
from typing import List, Any, Optional, Union, Dict from typing import List, Any, Optional, Union, Dict
from pydantic import BaseModel from pydantic import BaseModel
from lightrag import LightRAG, QueryParam from lightrag import LightRAG, QueryParam
from lightrag.types import GPTKeywordExtractionFormat
from lightrag.api import __api_version__ from lightrag.api import __api_version__
from lightrag.utils import EmbeddingFunc from lightrag.utils import EmbeddingFunc
from enum import Enum from enum import Enum
@@ -756,6 +757,9 @@ def create_app(args):
keyword_extraction=False, keyword_extraction=False,
**kwargs, **kwargs,
) -> str: ) -> str:
keyword_extraction = kwargs.pop("keyword_extraction", None)
if keyword_extraction:
kwargs["response_format"] = GPTKeywordExtractionFormat
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []
return await openai_complete_if_cache( return await openai_complete_if_cache(
@@ -775,6 +779,9 @@ def create_app(args):
keyword_extraction=False, keyword_extraction=False,
**kwargs, **kwargs,
) -> str: ) -> str:
keyword_extraction = kwargs.pop("keyword_extraction", None)
if keyword_extraction:
kwargs["response_format"] = GPTKeywordExtractionFormat
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []
return await azure_openai_complete_if_cache( return await azure_openai_complete_if_cache(