diff --git a/int_tests/test_chat.py b/int_tests/test_chat.py index ef72e85..c33c1cf 100644 --- a/int_tests/test_chat.py +++ b/int_tests/test_chat.py @@ -1,4 +1,3 @@ -import time import unittest from datetime import timedelta from pathlib import Path @@ -24,7 +23,7 @@ def setUp(self): self.search_params = SearchCorporaParameters( corpora=[ KeyedSearchCorpus( - corpus_key=self.key, + corpus_key="int-test-upload-fern", lexical_interpolation=0, ) ], @@ -41,7 +40,7 @@ def setUp(self): citations=CitationParameters( style="none", ), - enable_factual_consistency_score=True, + enable_factual_consistency_score=False, ) self.chat_params = ChatParameters(store=True) @@ -53,7 +52,7 @@ def setUp(self): def test_chat(self): response = self.client.chat( - query="what is vectara?", + query="Robot Utility Models", search=self.search_params, generation=self.generation_params, chat=self.chat_params diff --git a/src/vectara/client.py b/src/vectara/client.py index db7be74..66353a0 100644 --- a/src/vectara/client.py +++ b/src/vectara/client.py @@ -101,7 +101,7 @@ def chat( self, *, query: str, - search: typing.Optional[SearchCorporaParameters] = OMIT, + search: SearchCorporaParameters = OMIT, request_timeout: typing.Optional[int] = None, request_timeout_millis: typing.Optional[int] = None, generation: typing.Optional[GenerationParameters] = OMIT, @@ -136,8 +136,8 @@ def chat( ChatFullResponse """ - if not chat_id and not search: - raise ValueError("`search` parameter is required when `chat_id` is not provided.") + # if not chat_id and not search: + # raise ValueError("`search` parameter is required when `chat_id` is not provided.") if chat_id: session = self.session_manager.get_session(chat_id)