chore: Update Python API code to use optional typing and fix formatting

src/frontend/src/modals/apiModal/utils/get-python-api-code.tsx

@@ -8,51 +8,51 @@
 export default function getPythonApiCode(
   flowId: string,
   isAuth: boolean,
-  tweaksBuildedObject,
+  tweaksBuildedObject
 ): string {
   const tweaksObject = tweaksBuildedObject[0];
   return `import requests
-  from typing import Optional
+from typing import Optional
 
-  BASE_API_URL = "${window.location.protocol}//${window.location.host}/api/v1/run"
+BASE_API_URL = "${window.location.protocol}//${window.location.host}/api/v1/run"
-  FLOW_ID = "${flowId}"
+FLOW_ID = "${flowId}"
-  # You can tweak the flow by adding a tweaks dictionary
+# You can tweak the flow by adding a tweaks dictionary
-  # e.g {"OpenAI-XXXXX": {"model_name": "gpt-4"}}
+# e.g {"OpenAI-XXXXX": {"model_name": "gpt-4"}}
-  TWEAKS = ${JSON.stringify(tweaksObject, null, 2)}
+TWEAKS = ${JSON.stringify(tweaksObject, null, 2)}
 
-  def run_flow(message: str,
+def run_flow(message: str,
-    flow_id: str,
+  flow_id: str,
-    output_type: str = "chat",
+  output_type: str = "chat",
-    input_type: str = "chat",
+  input_type: str = "chat",
-    tweaks: Optional[dict] = None,
+  tweaks: Optional[dict] = None,
-    api_key: Optional[str] = None) -> dict:
+  api_key: Optional[str] = None) -> dict:
-      """
+    """
-      Run a flow with a given message and optional tweaks.
+    Run a flow with a given message and optional tweaks.
 
-      :param message: The message to send to the flow
+    :param message: The message to send to the flow
-      :param flow_id: The ID of the flow to run
+    :param flow_id: The ID of the flow to run
-      :param tweaks: Optional tweaks to customize the flow
+    :param tweaks: Optional tweaks to customize the flow
-      :return: The JSON response from the flow
+    :return: The JSON response from the flow
-      """
+    """
-      api_url = f"{BASE_API_URL}/{flow_id}"
+    api_url = f"{BASE_API_URL}/{flow_id}"
 
-      payload = {
+    payload = {
-          "input_value": message,
+        "input_value": message,
-          "output_type": output_type,
+        "output_type": output_type,
-          "input_type": input_type,
+        "input_type": input_type,
-      }
+    }
-      headers = None
+    headers = None
-      if tweaks:
+    if tweaks:
-          payload["tweaks"] = tweaks
+        payload["tweaks"] = tweaks
-      if api_key:
+    if api_key:
-          headers = {"x-api-key": api_key}
+        headers = {"x-api-key": api_key}
-      response = requests.post(api_url, json=payload, headers=headers)
+    response = requests.post(api_url, json=payload, headers=headers)
-      return response.json()
+    return response.json()
 
-  # Setup any tweaks you want to apply to the flow
+# Setup any tweaks you want to apply to the flow
-  message = "message"
+message = "message"
-  ${!isAuth ? `api_key = "<your api key>"` : ""}
+${!isAuth ? `api_key = "<your api key>"` : ""}
-  print(run_flow(message=message, flow_id=FLOW_ID, tweaks=TWEAKS${
+print(run_flow(message=message, flow_id=FLOW_ID, tweaks=TWEAKS${
     !isAuth ? `, api_key=api_key` : ""
   }))`;
 }

src/frontend/src/modals/apiModal/utils/get-python-code.tsx

@@ -6,15 +6,15 @@
  */
 export default function getPythonCode(
   flowName: string,
-  tweaksBuildedObject,
+  tweaksBuildedObject
 ): string {
   const tweaksObject = tweaksBuildedObject[0];
 
   return `from langflow.load import run_flow_from_json
-  TWEAKS = ${JSON.stringify(tweaksObject, null, 2)}
+TWEAKS = ${JSON.stringify(tweaksObject, null, 2)}
 
-  result = run_flow_from_json(flow="${flowName}.json",
+result = run_flow_from_json(flow="${flowName}.json",
-                              input_value="message",
+                            input_value="message",
-                              fallback_to_env_vars=True, # False by default
+                            fallback_to_env_vars=True, # False by default
-                              tweaks=TWEAKS)`;
+                            tweaks=TWEAKS)`;
 }