sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

Make online results an optional argument to the gpt converse method

Browse source
This commit is contained in:
Debanjum Singh Solanky 2023-12-03 19:13:28 -05:00
parent 7009793170
commit 2b09caa237
3 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -123,8 +123,8 @@ def send_message_to_model(
def converse( def converse(
references, references,
online_results,
user_query, user_query,
online_results=[],
conversation_log={}, conversation_log={},
model: str = "gpt-3.5-turbo", model: str = "gpt-3.5-turbo",
api_key: Optional[str] = None, api_key: Optional[str] = None,

4
src/khoj/routers/helpers.py
View file

@ -251,9 +251,9 @@ def generate_chat_response(
chat_model = conversation_config.chat_model chat_model = conversation_config.chat_model
chat_response = converse( chat_response = converse(
compiled_references, compiled_references,
online_results,
q, q,
meta_log, online_results=online_results,
conversation_log=meta_log,
model=chat_model, model=chat_model,
api_key=api_key, api_key=api_key,
completion_func=partial_completion, completion_func=partial_completion,

4
tests/test_openai_chat_director.py
View file

@ -68,10 +68,10 @@ def test_chat_with_online_content(chat_client):
response_message = response_message.split("### compiled references")[0] response_message = response_message.split("### compiled references")[0]
# Assert # Assert
expected_responses = ["http://www.paulgraham.com/greatwork.html"] expected_responses = ["http://www.paulgraham.com/greatwork.html", "Please set your SERPER_DEV_API_KEY"]
assert response.status_code == 200 assert response.status_code == 200
assert any([expected_response in response_message for expected_response in expected_responses]), ( assert any([expected_response in response_message for expected_response in expected_responses]), (
"Expected assistants name, [K|k]hoj, in response but got: " + response_message "Expected links or serper not setup in response but got: " + response_message
) )