Fix names

weaviate · Aug 29, 2024 · fce0476 · fce0476
1 parent 1cb1230
commit fce0476
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/test/collection/test_config.py b/test/collection/test_config.py
@@ -615,8 +615,17 @@ def test_config_with_vectorizer_and_properties(
         {"generative-anyscale": {}},
     ),
     (
-        Configure.Generative.friendliai(temperature=0.5, max_tokens=100, model="model", base_url="https://api.custom.ai"),
-        {"generative-friendliai": {"temperature": 0.5, "maxTokens": 100, "model": "model", "baseURL": "https://api.custom.ai"}},
+        Configure.Generative.friendliai(
+            temperature=0.5, max_tokens=100, model="model", base_url="https://api.custom.ai"
+        ),
+        {
+            "generative-friendliai": {
+                "temperature": 0.5,
+                "maxTokens": 100,
+                "model": "model",
+                "baseURL": "https://api.custom.ai",
+            }
+        },
     ),
     (
         Configure.Generative.mistral(temperature=0.5, max_tokens=100, model="model"),

diff --git a/weaviate/collections/classes/config.py b/weaviate/collections/classes/config.py
@@ -178,7 +178,7 @@ class GenerativeSearches(str, Enum):
     ANTHROPIC = "generative-anthropic"
     ANYSCALE = "generative-anyscale"
     COHERE = "generative-cohere"
-    FRIENDLYAI = "generative-friendlyai"
+    FRIENDLIAI = "generative-friendliai"
     MISTRAL = "generative-mistral"
     OCTOAI = "generative-octoai"
     OLLAMA = "generative-ollama"
@@ -447,7 +447,7 @@ class _GenerativeMistral(_GenerativeConfigCreate):
 
 class _GenerativeFriendlyai(_GenerativeConfigCreate):
     generative: Union[GenerativeSearches, _EnumLikeStr] = Field(
-        default=GenerativeSearches.FRIENDLYAI, frozen=True, exclude=True
+        default=GenerativeSearches.FRIENDLIAI, frozen=True, exclude=True
     )
     temperature: Optional[float]
     model: Optional[str]
@@ -636,7 +636,9 @@ def friendliai(
         temperature: Optional[float] = None,
         max_tokens: Optional[int] = None,
     ) -> _GenerativeConfigCreate:
-        return _GenerativeFriendlyai(model=model, temperature=temperature, maxTokens=max_tokens, baseURL=base_url)
+        return _GenerativeFriendlyai(
+            model=model, temperature=temperature, maxTokens=max_tokens, baseURL=base_url
+        )
 
     @staticmethod
     def mistral(