init fireworks

Author: cosminacho

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

@@ -7,16 +7,15 @@
 try:
     from langchain_fireworks.chat_models import ChatFireworks
 
-    from fireworks.client import AsyncFireworks, Fireworks
     from fireworks.client.api_client import FireworksClient as FireworksClientV1
 except ImportError as e:
     raise ImportError(
         "The 'fireworks' extra is required to use UiPathChatFireworks. "
-        "Install it with: uv add uipath-langchain-client[fireworks]"
+        'Install it with: uv add "uipath-langchain-client[fireworks]"'
     ) from e
 
 
-class UiPathChatFireworks(UiPathBaseLLMClient, ChatFireworks):
+class UiPathChatFireworks(UiPathBaseLLMClient, ChatFireworks):  # type: ignore[override]
     api_config: UiPathAPIConfig = UiPathAPIConfig(
         api_type="completions",
         client_type="passthrough",
@@ -35,6 +34,5 @@ def setup_uipath_client(self) -> Self:
         )
         fireworks_client_v1._client = self.uipath_sync_client
         fireworks_client_v1._async_client = self.uipath_async_client
-        self.client = Fireworks(client=fireworks_client_v1)
-
+        self.client._client_v1 = fireworks_client_v1
         return self

packages/uipath_langchain_client/src/uipath_langchain_client/clients/fireworks/embeddings.py

@@ -10,15 +10,17 @@
 except ImportError as e:
     raise ImportError(
         "The 'fireworks' extra is required to use UiPathFireworksEmbeddings. "
-        "Install it with: uv add uipath-langchain-client[fireworks]"
+        'Install it with: uv add "uipath-langchain-client[fireworks]"'
     ) from e
 
 
 class UiPathFireworksEmbeddings(UiPathBaseLLMClient, FireworksEmbeddings):
     api_config: UiPathAPIConfig = UiPathAPIConfig(
         api_type="embeddings",
         client_type="passthrough",
-        vendor_type="fireworks",
+        vendor_type="openai",
+        api_flavor="chat-completions",
+        api_version="2025-03-01-preview",
         freeze_base_url=True,
     )
 
@@ -27,13 +29,13 @@ def setup_uipath_client(self) -> Self:
         self.client = OpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_sync_client,
         )
         self.async_client = AsyncOpenAI(
             api_key="PLACEHOLDER",
             timeout=None,  # handled by the UiPath client
-            max_retries=1,  # handled by the UiPath client
+            max_retries=0,  # handled by the UiPath client
             http_client=self.uipath_async_client,
         )
         return self