🐛 FIX: Timeout parameter

Author: arre-ankit

examples/agent/agent.run.mcp.py

@@ -24,7 +24,7 @@ def main():
         exit(1)
 
     # Initialize Langbase client
-    langbase = Langbase(api_key=langbase_api_key, timeout=500)
+    langbase = Langbase(api_key=langbase_api_key)
 
     # Run the agent with MCP server
     response = langbase.agent_run(

langbase/langbase.py

@@ -31,8 +31,7 @@ class Langbase:
     def __init__(
         self,
         api_key: Optional[str] = None,
-        base_url: str = "https://api.langbase.com",
-        timeout: int = 30
+        base_url: str = "https://api.langbase.com"
     ):
         """
         Initialize the Langbase client.
@@ -41,7 +40,6 @@ def __init__(
             api_key: The API key for authentication. If not provided, it will be read
                     from the LANGBASE_API_KEY environment variable.
             base_url: The base URL for the API.
-            timeout: The timeout for API requests in seconds.
 
         Raises:
             ValueError: If no API key is provided and LANGBASE_API_KEY is not set.
@@ -53,12 +51,10 @@ def __init__(
             )
 
         self.base_url = base_url
-        self.timeout = timeout
 
         self.request = Request({
             "api_key": self.api_key,
-            "base_url": self.base_url,
-            "timeout": self.timeout
+            "base_url": self.base_url
         })
 
         # Initialize properties and methods
@@ -162,8 +158,7 @@ def run(
                 if api_key:
                     request = Request({
                         "api_key": api_key,
-                        "base_url": self.parent.base_url,
-                        "timeout": self.parent.timeout
+                        "base_url": self.parent.base_url
                     })
 
                 headers = {}
@@ -679,8 +674,7 @@ def parser(
         response = requests.post(
             f"{self.base_url}/v1/parser",
             headers={"Authorization": f"Bearer {self.api_key}"},
-            files=files,
-            timeout=self.timeout
+            files=files
         )
 
         if not response.ok:

langbase/request.py

@@ -29,12 +29,10 @@ def __init__(self, config: Dict[str, Any]):
             config: Configuration dictionary containing:
                 - api_key: API key for authentication
                 - base_url: Base URL for the API
-                - timeout: Timeout for requests in seconds (default: 30)
         """
         self.config = config
         self.api_key = config.get("api_key", "")
         self.base_url = config.get("base_url", "")
-        self.timeout = config.get("timeout", 30)
 
     def build_url(self, endpoint: str) -> str:
         """
@@ -107,7 +105,6 @@ def make_request(
                     url=url,
                     headers={k: v for k, v in headers.items() if k != 'Content-Type'},
                     files=files,
-                    timeout=self.timeout,
                     stream=stream
                 )
             else:
@@ -116,7 +113,6 @@ def make_request(
                     url=url,
                     headers=headers,
                     json=body if body else None,
-                    timeout=self.timeout,
                     stream=stream
                 )
             return response

langbase/types.py

@@ -467,7 +467,6 @@ class LangbaseOptions(TypedDict, total=False):
     """Options for initializing Langbase client."""
     api_key: str
     base_url: Literal['https://api.langbase.com', 'https://eu-api.langbase.com']
-    timeout: int
 
 
 # Protocol for file-like objects

tests/test_langbase.py

@@ -21,7 +21,6 @@ def test_initialization_with_api_key(self):
         """Test initialization with API key parameter."""
         self.assertEqual(self.lb.api_key, self.api_key)
         self.assertEqual(self.lb.base_url, "https://api.langbase.com")
-        self.assertEqual(self.lb.timeout, 30)
 
     @patch.dict(os.environ, {"LANGBASE_API_KEY": "env-api-key"}, clear=True)
     def test_initialization_with_env_var(self):

tests/test_request.py

@@ -20,16 +20,14 @@ def setUp(self):
         """Set up test fixtures."""
         self.config = {
             "api_key": "test-api-key",
-            "base_url": "https://api.langbase.com",
-            "timeout": 30
+            "base_url": "https://api.langbase.com"
         }
         self.request = Request(self.config)
 
     def test_initialization(self):
         """Test initialization."""
         self.assertEqual(self.request.api_key, "test-api-key")
         self.assertEqual(self.request.base_url, "https://api.langbase.com")
-        self.assertEqual(self.request.timeout, 30)
 
     def test_build_url(self):
         """Test build_url method."""
@@ -66,7 +64,6 @@ def test_make_request(self, mock_request):
             url="https://api.langbase.com/test",
             headers={"Authorization": "Bearer test-api-key"},
             json=None,
-            timeout=30,
             stream=False
         )
         self.assertEqual(response, mock_response)