yangdx
commited on
Commit
·
e8f9a30
1
Parent(s):
a6f5190
fix linting
Browse files- lightrag/api/lightrag_server.py +1 -4
- lightrag/operate.py +3 -1
lightrag/api/lightrag_server.py
CHANGED
|
@@ -725,10 +725,7 @@ def create_app(args):
|
|
| 725 |
from lightrag.llm.ollama import ollama_model_complete, ollama_embed
|
| 726 |
if args.llm_binding == "openai" or args.embedding_binding == "openai":
|
| 727 |
from lightrag.llm.openai import openai_complete_if_cache, openai_embed
|
| 728 |
-
if
|
| 729 |
-
args.llm_binding == "azure_openai"
|
| 730 |
-
or args.embedding_binding == "azure_openai"
|
| 731 |
-
):
|
| 732 |
from lightrag.llm.azure_openai import (
|
| 733 |
azure_openai_complete_if_cache,
|
| 734 |
azure_openai_embed,
|
|
|
|
| 725 |
from lightrag.llm.ollama import ollama_model_complete, ollama_embed
|
| 726 |
if args.llm_binding == "openai" or args.embedding_binding == "openai":
|
| 727 |
from lightrag.llm.openai import openai_complete_if_cache, openai_embed
|
| 728 |
+
if args.llm_binding == "azure_openai" or args.embedding_binding == "azure_openai":
|
|
|
|
|
|
|
|
|
|
| 729 |
from lightrag.llm.azure_openai import (
|
| 730 |
azure_openai_complete_if_cache,
|
| 731 |
azure_openai_embed,
|
lightrag/operate.py
CHANGED
|
@@ -1543,7 +1543,9 @@ async def naive_query(
|
|
| 1543 |
|
| 1544 |
sys_prompt_temp = PROMPTS["naive_rag_response"]
|
| 1545 |
sys_prompt = sys_prompt_temp.format(
|
| 1546 |
-
content_data=section,
|
|
|
|
|
|
|
| 1547 |
)
|
| 1548 |
|
| 1549 |
if query_param.only_need_prompt:
|
|
|
|
| 1543 |
|
| 1544 |
sys_prompt_temp = PROMPTS["naive_rag_response"]
|
| 1545 |
sys_prompt = sys_prompt_temp.format(
|
| 1546 |
+
content_data=section,
|
| 1547 |
+
response_type=query_param.response_type,
|
| 1548 |
+
history=history_context,
|
| 1549 |
)
|
| 1550 |
|
| 1551 |
if query_param.only_need_prompt:
|