feat: add base url for OpenAI connection (#166)

dynamiq/connections/connections.py

@@ -187,8 +187,10 @@ class OpenAI(BaseApiKeyConnection):
 
     Attributes:
         api_key (str): The API key for the OpenAI service, fetched from the environment variable 'OPENAI_API_KEY'.
+        url (str): The endpoint url for the OpenAI service, fetched from the environment variable 'OPENAI_URL'.
     """
     api_key: str = Field(default_factory=partial(get_env_var, "OPENAI_API_KEY"))
+    url: str = Field(default_factory=partial(get_env_var, "OPENAI_URL", "https://api.openai.com/v1"))
 
     def connect(self) -> "OpenAIClient":
         """
@@ -201,10 +203,21 @@ def connect(self) -> "OpenAIClient":
         """
         # Import in runtime to save memory
         from openai import OpenAI as OpenAIClient
-        openai_client = OpenAIClient(api_key=self.api_key)
+
+        openai_client = OpenAIClient(api_key=self.api_key, base_url=self.url)
         logger.debug("Connected to OpenAI")
         return openai_client
 
+    @property
+    def conn_params(self) -> dict:
+        """
+        Returns the parameters required for connection.
+        """
+        return {
+            "api_base": self.url,
+            "api_key": self.api_key,
+        }
+
 
 class Anthropic(BaseApiKeyConnection):
     api_key: str = Field(default_factory=partial(get_env_var, "ANTHROPIC_API_KEY"))

tests/integration/flows/test_flow.py

@@ -123,6 +123,7 @@ def test_workflow_with_depend_nodes_with_tracing(
             client=ANY,
             response_format=None,
             drop_params=True,
+            api_base="https://api.openai.com/v1",
         ),
         mock.call(
             tools=None,
@@ -277,6 +278,7 @@ def test_workflow_with_depend_nodes_and_depend_fail(
             top_p=None,
             response_format=None,
             drop_params=True,
+            api_base="https://api.openai.com/v1",
         )
     ]