tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-18 10:14:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: tests

Browse Source
This commit is contained in:
Alex committed 2024-12-20 18:13:37 +00:00
1 parent c2a95b5bec
commit 1f75f0c082
5 files changed
+21 -17
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/llm/test_sagemaker.py
+1 -1
View File
@@ -76,7 +76,7 @@ class TestSagemakerAPILLM(unittest.TestCase):
with patch.object(self.sagemaker.runtime, 'invoke_endpoint_with_response_stream',
return_value=self.response) as mock_invoke_endpoint:
output = list(self.sagemaker.gen_stream(None, self.messages))
output = list(self.sagemaker.gen_stream(None, self.messages, tools=None))
mock_invoke_endpoint.assert_called_once_with(
EndpointName=self.sagemaker.endpoint,
ContentType='application/json',