👌 IMPROVE: lint

Author: saqibameen

CONTRIBUTING.md

@@ -79,7 +79,7 @@ pytest -v
 
 ### 5. Run All Checks at Once
 ```bash
-# This runs all pre-commit hooks (black, isort, ruff, mypy)
+# This runs all pre-commit hooks (black, isort)
 pre-commit run --all-files
 ```
 

langbase/constants.py

@@ -26,6 +26,7 @@
     429: "RateLimitError",
 }
 
+BASE_URL = "https://api.langbase.com"
 # API Endpoints
 PIPES_ENDPOINT = "/v1/pipes"
 PIPE_DETAIL_ENDPOINT = "/v1/pipes/{name}"

langbase/langbase.py

@@ -31,8 +31,7 @@ def __init__(self, api_key: str = "", base_url: str = "https://api.langbase.com"
         Initialize the Langbase client.
 
         Args:
-            api_key: The API key for authentication. If not provided, it will be read
-                    from the LANGBASE_API_KEY environment variable.
+            api_key: The API key for authentication.
             base_url: The base URL for the API.
         """
         self.base_url = base_url

langbase/primitives/parser.py

@@ -47,15 +47,20 @@ def parser(
         Returns:
             Dictionary with document name and extracted content
         """
-        files = convert_document_to_request_files(document, document_name, content_type)
+        document_content = convert_document_to_request_files(
+            document, document_name, content_type
+        )
 
-        response = requests.post(
-            f"{self.parent.base_url}{PARSER_ENDPOINT}",
+        response = self.request.post(
+            PARSER_ENDPOINT,
             headers={"Authorization": f"Bearer {self.parent.api_key}"},
-            files=files,
+            document=document_content,
         )
 
-        if not response.ok:
-            self.request.handle_error_response(response)
+        # Transform API response: rename documentName to document_name
+        if isinstance(response, dict) and "documentName" in response:
+            response["document_name"] = response.pop("documentName")
+
+        print("response", response)
 
-        return response.json()
+        return response

langbase/primitives/threads.py

@@ -79,7 +79,7 @@ def update(self, thread_id: str, metadata: Dict[str, str]) -> ThreadsBaseRespons
         Returns:
             Updated thread object
         """
-        options = {"threadId": thread_id, "metadata": metadata}
+        options = {"metadata": metadata}
         return self.request.post(
             THREAD_DETAIL_ENDPOINT.format(thread_id=thread_id), options
         )
@@ -121,8 +121,10 @@ def append(
         Returns:
             List of added messages
         """
+        options = {"messages": messages}
+
         return self.request.post(
-            THREAD_MESSAGES_ENDPOINT.format(thread_id=thread_id), messages
+            THREAD_MESSAGES_ENDPOINT.format(thread_id=thread_id), options
         )
 
     def list(self, thread_id: str) -> List[ThreadMessagesBaseResponse]:

langbase/request.py

@@ -316,7 +316,7 @@ def post(
         body: Optional[Dict[str, Any]] = None,
         headers: Optional[Dict[str, str]] = None,
         stream: bool = False,
-        files: Optional[Dict[str, Any]] = None,
+        document: Optional[Dict[str, Any]] = None,
     ) -> Any:
         """
         Send a POST request to the API.
@@ -331,7 +331,7 @@ def post(
         Returns:
             Processed API response
         """
-        return self.send(endpoint, "POST", headers, body, stream, files)
+        return self.send(endpoint, "POST", headers, body, stream, document)
 
     def get(
         self,

langbase/types.py

@@ -83,7 +83,7 @@ class Message(TypedDict, total=False):
     """Basic message structure."""
 
     role: Role
-    content: string | NULL
+    content: Optional[str]
     name: Optional[str]
     tool_call_id: Optional[str]
     tool_calls: Optional[List[ToolCall]]

requirements-dev.txt

@@ -11,13 +11,6 @@ responses>=0.23.0
 black>=22.1.0
 isort>=5.10.1
 
-# Type checking
-mypy>=1.0.0
-types-requests>=2.28.0
-
-# Linting
-ruff>=0.1.0
-
 # Pre-commit hooks
 pre-commit>=3.0.0
 
@@ -27,4 +20,4 @@ twine>=4.0.0
 
 # Development utilities
 ipdb>=0.13.0
-python-dotenv>=0.19.0
+python-dotenv>=0.19.0

tests/conftest.py

@@ -186,11 +186,33 @@ def mock_responses():
             "created_at": timestamp,
             "metadata": {},
         },
+        "threads_create_with_metadata": {
+            "id": "thread_123",
+            "object": "thread",
+            "created_at": timestamp,
+            "metadata": {"user_id": "123", "session": "abc"},
+        },
+        "threads_create_with_thread_id": {
+            "id": "custom_thread_456",
+            "object": "thread",
+            "created_at": timestamp,
+            "metadata": {},
+        },
+        "threads_create_with_messages": {
+            "id": "thread_123",
+            "object": "thread",
+            "created_at": timestamp,
+            "metadata": {},
+            "messages": [
+                {"role": "user", "content": "Hello"},
+                {"role": "assistant", "content": "Hi there!"},
+            ],
+        },
         "threads_update": {
             "id": "thread_123",
             "object": "thread",
             "created_at": timestamp,
-            "metadata": {"updated": "true"},
+            "metadata": {"user_id": "123", "session": "abc"},
         },
         "threads_get": {
             "id": "thread_123",
@@ -244,7 +266,7 @@ def mock_responses():
         "embed": [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]],
         "chunker": ["First chunk", "Second chunk", "Third chunk"],
         "parser": {
-            "document_name": "test.pdf",
+            "documentName": "test.pdf",
             "content": "Parsed document content from test.pdf",
         },
         # Agent run response (similar to pipe run)

tests/constants.py

@@ -0,0 +1,12 @@
+AUTHORIZATION_HEADER = {
+    "Authorization": "Bearer test-api-key",
+}
+
+JSON_CONTENT_TYPE_HEADER = {
+    "Content-Type": "application/json",
+}
+
+AUTH_AND_JSON_CONTENT_HEADER = {
+    **AUTHORIZATION_HEADER,
+    **JSON_CONTENT_TYPE_HEADER,
+}