feat: claude api support (#572)

2023-07-17 00:14:19 +08:00
parent 510389909c
commit 7599f79a17
52 changed files with 637 additions and 349 deletions
--- a/api/services/hit_testing_service.py
+++ b/api/services/hit_testing_service.py
@@ -31,7 +31,7 @@ class HitTestingService:

        model_credentials = LLMBuilder.get_model_credentials(
            tenant_id=dataset.tenant_id,
-            model_provider=LLMBuilder.get_default_provider(dataset.tenant_id),
+            model_provider=LLMBuilder.get_default_provider(dataset.tenant_id, 'text-embedding-ada-002'),
            model_name='text-embedding-ada-002'
        )