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

Improve Inferred Document Search Query Extraction from GPT

Browse Source 
Using stop_words = "\n" was preventing JSON responses with newlines in
them
This commit is contained in:
Debanjum Singh Solanky
2024-04-11 19:21:08 +05:30
parent 653681967e
commit 15a78b19ad
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/processor/conversation/openai/gpt.py
View File

@@ -67,7 +67,7 @@ def extract_questions(
model_name=model, model_name=model,
temperature=temperature, temperature=temperature,
max_tokens=max_tokens, max_tokens=max_tokens,
model_kwargs={"stop": ["A: ", "\n"], "response_format": {"type": "json_object"}}, model_kwargs={"response_format": {"type": "json_object"}},
openai_api_key=api_key, openai_api_key=api_key,
) )