test

Author: StpMax

tests/test_sdk.py

@@ -1301,16 +1301,6 @@ def check_project_kb(self, project, database, mock_get, mock_del, mock_post, moc
             },
         }}}
 
-        # completion
-        kb.completion('hi', type='chat', llm_model='gpt-4')
-        args, kwargs = mock_post.call_args
-        assert args[0] == f'{DEFAULT_CLOUD_API_URL}/api/projects/{project.name}/knowledge_bases/{kb.name}/completions'
-        assert kwargs == {'json': {
-            'query': 'hi',
-            'type': 'chat',
-            'llm_model': 'gpt-4',
-        }}
-
         # drop
         project.knowledge_bases.drop('kb2')