fix: add ANTHROPIC/AZURE cases to validator + remove unused original_message param

- clients/anthropic/chat_models.py: add VendorType.ANTHROPIC and
  VendorType.AZURE cases to setup_api_flavor_and_version validator so the
  class can be constructed with all supported vendor types without raising
- clients/normalized/chat_models.py: remove the now-unused original_message
  parameter from _generate_chunk and update both call sites in
  _uipath_stream and _uipath_astream

Author: cosminacho

packages/uipath_langchain_client/src/uipath_langchain_client/clients/anthropic/chat_models.py

@@ -45,15 +45,17 @@ class UiPathChatAnthropic(UiPathBaseChatModel, ChatAnthropic):
     def setup_api_flavor_and_version(self) -> Self:
         self.api_config.vendor_type = self.vendor_type
         match self.vendor_type:
+            case VendorType.ANTHROPIC:
+                self.api_config.api_flavor = ApiFlavor.CHAT_COMPLETIONS
+            case VendorType.AZURE:
+                self.api_config.api_flavor = ApiFlavor.CHAT_COMPLETIONS
             case VendorType.VERTEXAI:
                 self.api_config.api_flavor = ApiFlavor.ANTHROPIC_CLAUDE
                 self.api_config.api_version = "v1beta1"
             case VendorType.AWSBEDROCK:
                 self.api_config.api_flavor = ApiFlavor.INVOKE
             case _:
-                raise ValueError(
-                    "anthropic and azure vendors are currently not supported by UiPath"
-                )
+                raise ValueError(f"Unsupported vendor_type: {self.vendor_type}")
         return self
 
     # Override fields to avoid typing issues and fix stuff

packages/uipath_langchain_client/src/uipath_langchain_client/clients/normalized/chat_models.py

@@ -328,9 +328,7 @@ async def _uipath_agenerate(
         response = await self.uipath_arequest(request_body=request_body, raise_status_error=True)
         return self._postprocess_response(response.json())
 
-    def _generate_chunk(
-        self, original_message: str, json_data: dict[str, Any]
-    ) -> ChatGenerationChunk:
+    def _generate_chunk(self, json_data: dict[str, Any]) -> ChatGenerationChunk:
         generation_info = {
             "id": json_data.get("id"),
             "created": json_data.get("created", ""),
@@ -406,7 +404,7 @@ def _uipath_stream(
                 continue
             if "id" in json_data and not json_data["id"]:
                 continue
-            yield self._generate_chunk(chunk, json_data)
+            yield self._generate_chunk(json_data)
 
     async def _uipath_astream(
         self,
@@ -429,4 +427,4 @@ async def _uipath_astream(
                 continue
             if "id" in json_data and not json_data["id"]:
                 continue
-            yield self._generate_chunk(chunk, json_data)
+            yield self._generate_chunk(json_data)