🐛 FIX: Types

Author: arre-ankit

langbase/request.py

@@ -101,10 +101,14 @@ def make_request(
         try:
             # If files are provided, don't send JSON body
             if files:
+                # Remove Content-Type header for file uploads (requests will set it automatically)
+                filtered_headers = {
+                    k: v for k, v in headers.items() if k != "Content-Type"
+                }
                 response = requests.request(
                     method=method,
                     url=url,
-                    headers={k: v for k, v in headers.items() if k != "Content-Type"},
+                    headers=filtered_headers,
                     files=files,
                     stream=stream,
                 )
@@ -275,25 +279,28 @@ def send(
             thread_id = response.headers.get("lb-thread-id")
 
             if not body:
+                raw_response = body.get("raw_response", False) if body else False
                 return self.handle_run_response(
                     response,
                     thread_id=None,
-                    raw_response=body.get("raw_response", False) if body else False,
+                    raw_response=raw_response,
                     endpoint=endpoint,
                 )
 
             if body.get("stream") and "run" in url:
+                raw_response = body.get("raw_response", False)
                 return self.handle_run_response_stream(
-                    response, raw_response=body.get("raw_response", False)
+                    response, raw_response=raw_response
                 )
 
             if body.get("stream"):
                 return self.handle_stream_response(response)
 
+            raw_response = body.get("raw_response", False)
             return self.handle_run_response(
                 response,
                 thread_id=thread_id,
-                raw_response=body.get("raw_response", False),
+                raw_response=raw_response,
                 endpoint=endpoint,
             )
         # For non-generation endpoints, just return the JSON response

langbase/types.py

@@ -79,21 +79,11 @@ class ToolChoice(TypedDict):
     function: Dict[str, str]
 
 
-# Message types
-class MessageContentItem(TypedDict, total=False):
-    """Content item for a message with multiple content parts."""
-
-    type: str
-    text: Optional[str]
-    image_url: Optional[Dict[str, str]]
-    cache_control: Optional[Dict[str, str]]
-
-
 class Message(TypedDict, total=False):
     """Basic message structure."""
 
     role: Role
-    content: Optional[Union[str, List[MessageContentItem]]]
+    content: string | NULL
     name: Optional[str]
     tool_call_id: Optional[str]
     tool_calls: Optional[List[ToolCall]]
@@ -783,6 +773,3 @@ class AgentRunOptionsStreamT(TypedDict):
 
 # Agent response type (reuses RunResponse)
 AgentRunResponse = RunResponse
-
-
-# Workflow types - moved to workflow.py for better type support with generics