Fix test_anthropic_call_arg (#8975)

GitOrigin-RevId: ce182e17e0f99283f8798c4d227171bf89843a97

Author: szymondudycz

python/pathway/xpacks/llm/_utils.py

@@ -49,7 +49,7 @@ def wrapper(*args, **kwargs):
         return func
 
 
-def _check_model_accepts_arg(model_name: str, provider: str, arg: str):
+def _check_model_accepts_arg(model_name: str, provider: str | None, arg: str):
     from litellm import get_supported_openai_params
 
     supported_params = (

python/pathway/xpacks/llm/llms.py

@@ -442,10 +442,14 @@ def _accepts_call_arg(self, arg_name: str) -> bool:
         if self.model is None:
             return False
 
-        provider = self.model.split("/")[0]
-        model = "".join(
-            self.model.split("/")[1:]
-        )  # handle case: replicate/meta/meta-llama-3-8b
+        if "/" in self.model:
+            provider = self.model.split("/")[0]
+            model = "".join(
+                self.model.split("/")[1:]
+            )  # handle case: replicate/meta/meta-llama-3-8b
+        else:
+            model = self.model
+            provider = None
         return _check_model_accepts_arg(model, provider, arg_name)
 
 

python/pathway/xpacks/llm/tests/test_llms.py

@@ -97,12 +97,12 @@ def test_openai_call_args(model, call_arg):
     [
         "claude-3-5-sonnet-20240620",
         "claude-3-opus-20240229",
-        "antrophic/claude-3-5-sonnet-20240620",
+        "anthropic/claude-3-5-sonnet-20240620",
         None,
     ],
 )
 @pytest.mark.parametrize("call_arg", [*VALID_ARGS, *INVALID_ARGS])
-def test_antrophic_call_args(model, call_arg):
+def test_anthropic_call_args(model, call_arg):
     llm = llms.LiteLLMChat(model=model)
 
     if model is None: