klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 13:25:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:debanjum/khoj

Browse Source
This commit is contained in:
sabaimran
2023-07-10 14:31:19 -07:00
parent 53809298c0 5b37e988e6
commit 9a63f89f33
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/khoj/routers/api.py
View File

@@ -535,6 +535,7 @@ async def extract_references_and_questions(
# Initialize Variables # Initialize Variables
api_key = state.processor_config.conversation.openai_api_key api_key = state.processor_config.conversation.openai_api_key
chat_model = state.processor_config.conversation.chat_model
conversation_type = "general" if q.startswith("@general") else "notes" conversation_type = "general" if q.startswith("@general") else "notes"
compiled_references = [] compiled_references = []
inferred_queries = [] inferred_queries = []
@@ -542,7 +543,7 @@ async def extract_references_and_questions(
if conversation_type == "notes": if conversation_type == "notes":
# Infer search queries from user message # Infer search queries from user message
with timer("Extracting search queries took", logger): with timer("Extracting search queries took", logger):
inferred_queries = extract_questions(q, api_key=api_key, conversation_log=meta_log) inferred_queries = extract_questions(q, model=chat_model, api_key=api_key, conversation_log=meta_log)
# Collate search results as context for GPT # Collate search results as context for GPT
with timer("Searching knowledge base took", logger): with timer("Searching knowledge base took", logger):