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

Fix formatting of gpt.py

Browse Source
This commit is contained in:
sabaimran
2023-09-18 11:09:02 -07:00
parent 34b202b868
commit b225d1188c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -42,7 +42,12 @@ def summarize(session, model, api_key=None, temperature=0.5, max_tokens=200):
def extract_questions( def extract_questions(
text, model: Optional[str] = "gpt-4", conversation_log={}, api_key=None, temperature=0, max_tokens=100 text,
model: Optional[str] = "gpt-4",
conversation_log={},
api_key=None,
temperature=0,
max_tokens=100,
): ):
""" """
Infer search queries to retrieve relevant notes to answer user query Infer search queries to retrieve relevant notes to answer user query
@@ -70,7 +75,7 @@ def extract_questions(
bob_age={current_new_year.year - 1984}, bob_age={current_new_year.year - 1984},
chat_history=chat_history, chat_history=chat_history,
text=text, text=text,
yesterday_date=(today - timedelta(days=1)).strftime("%Y-%m-%d") yesterday_date=(today - timedelta(days=1)).strftime("%Y-%m-%d"),
) )
messages = [ChatMessage(content=prompt, role="assistant")] messages = [ChatMessage(content=prompt, role="assistant")]