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

Fix openai chat actor, director tests

Browse Source 
- Update test ChatModelOptions setup since update to it's schema
- Fix stale function calls using their updated signatures
This commit is contained in:
Debanjum Singh Solanky
2024-06-09 07:16:55 +05:30
parent f91cdf8e18
commit f440ddbe1d
4 changed files with 62 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_openai_chat_director.py
View File

@@ -516,7 +516,7 @@ async def test_get_correct_tools_online(chat_client):
user_query = "What's the weather in Patagonia this week?"
# Act
tools = await aget_relevant_information_sources(user_query, {})
tools = await aget_relevant_information_sources(user_query, {}, False)
# Assert
tools = [tool.value for tool in tools]
@@ -531,7 +531,7 @@ async def test_get_correct_tools_notes(chat_client):
user_query = "Where did I go for my first battleship training?"
# Act
tools = await aget_relevant_information_sources(user_query, {})
tools = await aget_relevant_information_sources(user_query, {}, False)
# Assert
tools = [tool.value for tool in tools]
@@ -546,7 +546,7 @@ async def test_get_correct_tools_online_or_general_and_notes(chat_client):
user_query = "What's the highest point in Patagonia and have I been there?"
# Act
tools = await aget_relevant_information_sources(user_query, {})
tools = await aget_relevant_information_sources(user_query, {}, False)
# Assert
tools = [tool.value for tool in tools]
@@ -563,7 +563,7 @@ async def test_get_correct_tools_general(chat_client):
user_query = "How many noble gases are there?"
# Act
tools = await aget_relevant_information_sources(user_query, {})
tools = await aget_relevant_information_sources(user_query, {}, False)
# Assert
tools = [tool.value for tool in tools]
@@ -587,7 +587,7 @@ async def test_get_correct_tools_with_chat_history(chat_client):
chat_history = generate_history(chat_log)
# Act
tools = await aget_relevant_information_sources(user_query, chat_history)
tools = await aget_relevant_information_sources(user_query, chat_history, False)
# Assert
tools = [tool.value for tool in tools]