retab 0.0.36__py3-none-any.whl → 0.0.38__py3-none-any.whl

{uiform → retab}/resources/consensus/responses_stream.py

@@ -1,33 +1,20 @@
-import json
-from pathlib import Path
-from typing import Any, AsyncGenerator, Generator, TypeVar, Generic, Optional, Union, List, Sequence, cast
-
-from openai.types.chat.chat_completion_reasoning_effort import ChatCompletionReasoningEffort
-from openai.types.chat.parsed_chat_completion import ParsedChatCompletionMessage
-from openai.types.responses.response import Response
-from openai.types.responses.response_input_param import ResponseInputParam, ResponseInputItemParam
-from openai.types.responses.response_output_item import ResponseOutputItem
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+from typing import AsyncGenerator, Generator, Optional, TypeVar, Union
+
+from openai.types.responses.response_input_param import ResponseInputParam
+from openai.types.responses.response_text_config_param import ResponseTextConfigParam
+from openai.types.shared_params.reasoning import Reasoning
 from pydantic import BaseModel
 
 from ..._resource import AsyncAPIResource, SyncAPIResource
 from ..._utils.ai_models import assert_valid_model_extraction
-from ..._utils.json_schema import load_json_schema, unflatten_dict
-from ..._utils.responses import convert_to_openai_format, convert_from_openai_format, parse_openai_responses_response
 from ..._utils.stream_context_managers import as_async_context_manager, as_context_manager
-from ...types.chat import ChatCompletionUiformMessage
-from ...types.completions import UiChatResponseCreateRequest, UiChatCompletionsRequest
-from ...types.documents.extractions import UiParsedChatCompletion, UiParsedChatCompletionChunk, UiParsedChoice, UiResponse
-from ...types.standards import PreparedRequest
+from ...types.completions import UiChatResponseCreateRequest
+from ...types.documents.extractions import UiResponse
 from ...types.schemas.object import Schema
+from ...types.standards import PreparedRequest
 
-from typing import Optional, Union
-from openai.types.shared_params.reasoning import Reasoning
-from openai.types.responses.response_input_param import ResponseInputParam
-from openai.types.responses.response_text_config_param import ResponseTextConfigParam
-from openai.types.shared_params.response_format_json_schema import ResponseFormatJSONSchema
+T = TypeVar("T", bound=BaseModel)
 
-T = TypeVar('T', bound=BaseModel)
 
 class BaseResponsesMixin:
     def prepare_create(
@@ -56,35 +43,20 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=stream,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
+        return PreparedRequest(method="POST", url="/v1/responses", data=request.model_dump(), idempotency_key=idempotency_key)
 
     def prepare_parse(
         self,
@@ -108,47 +80,29 @@ class BaseResponsesMixin:
 
         if instructions is None:
             instructions = schema_obj.developer_system_prompt
-        
+
         # Create the request object based on the UiChatResponseCreateRequest model
-        data = UiChatResponseCreateRequest(
+        request = UiChatResponseCreateRequest(
             model=model,
             input=input,
             temperature=temperature,
             stream=stream,
-            reasoning=reasoning, 
+            reasoning=reasoning,
             n_consensus=n_consensus,
-            text={
-                "format": {
-                    "type": "json_schema", 
-                    "name": schema_obj.id, 
-                    "schema": schema_obj.inference_json_schema, 
-                    "strict": True
-                }
-            },
+            text={"format": {"type": "json_schema", "name": schema_obj.id, "schema": schema_obj.inference_json_schema, "strict": True}},
             instructions=instructions,
         )
 
-        # Validate the request data
-        ui_chat_response_create_request = UiChatResponseCreateRequest.model_validate(data)
-
-        return PreparedRequest(
-            method="POST", 
-            url="/v1/responses", 
-            data=ui_chat_response_create_request.model_dump(), 
-            idempotency_key=idempotency_key
-        )
-        
-
         return PreparedRequest(
-            method="POST", 
-            url="/v1/completions", 
-            data=ui_chat_completions_request.model_dump(), 
-            idempotency_key=idempotency_key
+            method="POST",
+            url="/v1/responses",
+            data=request.model_dump(),
+            idempotency_key=idempotency_key,
         )
 
 
 class Responses(SyncAPIResource, BaseResponsesMixin):
-    """UiForm Responses API compatible with OpenAI Responses API"""
+    """Retab Responses API compatible with OpenAI Responses API"""
 
     @as_context_manager
     def stream(
@@ -163,8 +117,8 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
         idempotency_key: Optional[str] = None,
     ) -> Generator[UiResponse, None, None]:
         """
-        Create a completion using the UiForm API with streaming enabled.
-        
+        Create a completion using the Retab API with streaming enabled.
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -174,13 +128,13 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Generator[UiResponse]: Stream of responses
 
         Usage:
         ```python
-        with uiform.responses.stream(model, input, text, temperature, reasoning) as stream:
+        with retab.responses.stream(model, input, text, temperature, reasoning) as stream:
             for response in stream:
                 print(response)
         ```
@@ -217,8 +171,8 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
         idempotency_key: Optional[str] = None,
     ) -> Generator[UiResponse, None, None]:
         """
-        Parse content using the UiForm API with streaming enabled.
-        
+        Parse content using the Retab API with streaming enabled.
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -228,13 +182,13 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             Generator[UiResponse]: Stream of parsed responses
 
         Usage:
         ```python
-        with uiform.responses.stream_parse(model, input, MyModel, temperature, reasoning) as stream:
+        with retab.responses.stream_parse(model, input, MyModel, temperature, reasoning) as stream:
             for response in stream:
                 print(response)
         ```
@@ -259,9 +213,8 @@ class Responses(SyncAPIResource, BaseResponsesMixin):
             yield response
 
 
-
 class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
-    """UiForm Responses API compatible with OpenAI Responses API for async usage"""
+    """Retab Responses API compatible with OpenAI Responses API for async usage"""
 
     @as_async_context_manager
     async def stream(
@@ -276,8 +229,8 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
         idempotency_key: Optional[str] = None,
     ) -> AsyncGenerator[UiResponse, None]:
         """
-        Create a completion using the UiForm API asynchronously with streaming enabled.
-        
+        Create a completion using the Retab API asynchronously with streaming enabled.
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -287,13 +240,13 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             AsyncGenerator[UiResponse]: Async stream of responses
 
         Usage:
         ```python
-        async with uiform.responses.async_stream(model, input, text, temperature, reasoning) as stream:
+        async with retab.responses.async_stream(model, input, text, temperature, reasoning) as stream:
             async for response in stream:
                 print(response)
         ```
@@ -316,7 +269,7 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
                 continue
             response = UiResponse.model_validate(chunk_json)
             yield response
-        
+
     @as_async_context_manager
     async def stream_parse(
         self,
@@ -330,8 +283,8 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
         idempotency_key: Optional[str] = None,
     ) -> AsyncGenerator[UiResponse, None]:
         """
-        Parse content using the UiForm API asynchronously with streaming enabled.
-        
+        Parse content using the Retab API asynchronously with streaming enabled.
+
         Args:
             model: The model to use
             input: The input text or message array
@@ -341,13 +294,13 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
             n_consensus: Number of consensus models to use
             instructions: Optional system instructions
             idempotency_key: Idempotency key for request
-            
+
         Returns:
             AsyncGenerator[UiResponse]: Async stream of parsed responses
 
         Usage:
         ```python
-        async with uiform.responses.async_stream_parse(model, input, MyModel, temperature, reasoning) as stream:
+        async with retab.responses.async_stream_parse(model, input, MyModel, temperature, reasoning) as stream:
             async for response in stream:
                 print(response)
         ```
@@ -370,4 +323,3 @@ class AsyncResponses(AsyncAPIResource, BaseResponsesMixin):
                 continue
             response = UiResponse.model_validate(chunk_json)
             yield response
-