allow custom base_url of dify api server (#6510)

2024-07-22 13:24:24 +08:00
parent a6350daa02
commit 5b89b6fe2d
1 changed files with 2 additions and 2 deletions
--- a/sdks/python-client/dify_client/client.py
+++ b/sdks/python-client/dify_client/client.py
@@ -2,9 +2,9 @@ import requests


 class DifyClient:
-    def __init__(self, api_key):
+    def __init__(self, api_key, base_url: str = 'https://api.dify.ai/v1'):
        self.api_key = api_key
-        self.base_url = "https://api.dify.ai/v1"
+        self.base_url = base_url

    def _send_request(self, method, endpoint, json=None, params=None, stream=False):
        headers = {