fixed???

Author: tymekaxai

src/axiomatic/client.py

@@ -2,7 +2,7 @@
 import requests
 import os
 import time
-from typing import Dict, List
+from typing import Dict, List, Union, Optional
 
 from .base_client import BaseClient, AsyncBaseClient
 from . import ParseResponse
@@ -34,9 +34,9 @@ def create_report(self, response: EquationExtractionResponse, path: str):
 
     def analyze_equations(
         self,
-        file_path: str | None = None,
-        url_path: str | None = None
-    ) -> EquationExtractionResponse | None:
+        file_path: Optional[str] = None,
+        url_path: Optional[str] = None
+    ) -> Optional[EquationExtractionResponse]:
         if file_path:
             with open(file_path, "rb") as file:
                 response = self._ax_client.document.equation.from_pdf(document=file)
@@ -45,7 +45,7 @@ def analyze_equations(
             if "arxiv" in url_path and "abs" in url_path:
                 url_path = url_path.replace("abs", "pdf")
 
-            response = self._ax_client.document.equation.from_pdf(url=url_path)
+            response = self._ax_client.document.equation.from_pdf(document_url=url_path)
 
         else:
             print("Please provide either a file path or a URL to analyze.")