openaivec 0.12.5__py3-none-any.whl → 1.0.10__py3-none-any.whl

openaivec/task/__init__.py

@@ -32,7 +32,7 @@ Specialized tasks for customer service operations:
 ### Quick Start with Default Tasks
 ```python
 from openai import OpenAI
-from openaivec.responses import BatchResponses
+from openaivec import BatchResponses
 from openaivec.task import nlp, customer_support
 
 client = OpenAI()
@@ -90,15 +90,17 @@ results_df = df.ai.extract("sentiment")
 
 ### Spark Integration
 ```python
-from openaivec.spark import ResponsesUDFBuilder
+from openaivec.spark import task_udf
 
 # Register UDF for large-scale processing
 spark.udf.register(
     "analyze_sentiment",
-    ResponsesUDFBuilder.of_openai(
-        api_key=api_key,
-        model_name="gpt-4.1-mini"
-    ).build_from_task(task=nlp.SENTIMENT_ANALYSIS)
+    task_udf(
+        task=nlp.SENTIMENT_ANALYSIS,
+        model_name="gpt-4.1-mini",
+        batch_size=64,
+        max_concurrency=8,
+    ),
 )
 
 # Use in Spark SQL
@@ -117,7 +119,7 @@ All tasks are built using the `PreparedTask` dataclass:
 @dataclass(frozen=True)
 class PreparedTask:
     instructions: str           # Detailed prompt for the LLM
-    response_format: Type[ResponseFormat]    # Pydantic model or str for structured/plain output
+    response_format: type[ResponseFormat]    # Pydantic model or str for structured/plain output
     temperature: float = 0.0    # Sampling temperature
     top_p: float = 1.0         # Nucleus sampling parameter
 ```

openaivec/task/customer_support/__init__.py

@@ -1,32 +1,26 @@
 # Function imports
-from .inquiry_classification import inquiry_classification
-from .urgency_analysis import urgency_analysis
-from .customer_sentiment import customer_sentiment
-from .intent_analysis import intent_analysis
-from .inquiry_summary import inquiry_summary
-from .response_suggestion import response_suggestion
+from .customer_sentiment import CUSTOMER_SENTIMENT, customer_sentiment
 
 # Backward compatibility - constant imports
-from .inquiry_classification import INQUIRY_CLASSIFICATION
-from .urgency_analysis import URGENCY_ANALYSIS
-from .customer_sentiment import CUSTOMER_SENTIMENT
-from .intent_analysis import INTENT_ANALYSIS
-from .inquiry_summary import INQUIRY_SUMMARY
-from .response_suggestion import RESPONSE_SUGGESTION
+from .inquiry_classification import INQUIRY_CLASSIFICATION, inquiry_classification
+from .inquiry_summary import INQUIRY_SUMMARY, inquiry_summary
+from .intent_analysis import INTENT_ANALYSIS, intent_analysis
+from .response_suggestion import RESPONSE_SUGGESTION, response_suggestion
+from .urgency_analysis import URGENCY_ANALYSIS, urgency_analysis
 
 __all__ = [
     # Configurable functions (recommended)
     "inquiry_classification",
-    "urgency_analysis", 
+    "urgency_analysis",
     "customer_sentiment",
     "intent_analysis",
     "inquiry_summary",
     "response_suggestion",
     # Backward compatibility constants
     "INQUIRY_CLASSIFICATION",
-    "URGENCY_ANALYSIS", 
+    "URGENCY_ANALYSIS",
     "CUSTOMER_SENTIMENT",
     "INTENT_ANALYSIS",
     "INQUIRY_SUMMARY",
     "RESPONSE_SUGGESTION",
-]
+]

openaivec/task/customer_support/customer_sentiment.py

@@ -9,7 +9,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -49,7 +49,10 @@ Example:
 
     # Extract sentiment components
     extracted_df = df.ai.extract("sentiment")
-    print(extracted_df[["inquiry", "sentiment_satisfaction_level", "sentiment_churn_risk", "sentiment_emotional_state"]])
+    print(extracted_df[[
+        "inquiry", "sentiment_satisfaction_level",
+        "sentiment_churn_risk", "sentiment_emotional_state"
+    ]])
     ```
 
 Attributes:
@@ -58,11 +61,11 @@ Attributes:
         top_p=1.0 for deterministic output.
 """
 
-from typing import List, Literal
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["customer_sentiment"]
 
@@ -83,7 +86,7 @@ class CustomerSentiment(BaseModel):
     )
     sentiment_intensity: float = Field(description="Intensity of sentiment from 0.0 (mild) to 1.0 (extreme)")
     polarity_score: float = Field(description="Polarity score from -1.0 (very negative) to 1.0 (very positive)")
-    tone_indicators: List[str] = Field(description="Specific words or phrases indicating tone")
+    tone_indicators: list[str] = Field(description="Specific words or phrases indicating tone")
     relationship_status: Literal["new", "loyal", "at_risk", "detractor", "advocate"] = Field(
         description="Customer relationship status (new, loyal, at_risk, detractor, advocate)"
     )
@@ -92,21 +95,18 @@ class CustomerSentiment(BaseModel):
     )
 
 
-def customer_sentiment(
-    business_context: str = "general customer support", temperature: float = 0.0, top_p: float = 1.0
-) -> PreparedTask:
+def customer_sentiment(business_context: str = "general customer support") -> PreparedTask:
     """Create a configurable customer sentiment analysis task.
 
     Args:
         business_context (str): Business context for sentiment analysis.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
 
     Returns:
         PreparedTask configured for customer sentiment analysis.
     """
 
-    instructions = f"""Analyze customer sentiment in the context of support interactions, focusing on satisfaction, emotional state, and business implications.
+    instructions = f"""Analyze customer sentiment in the context of support interactions, focusing on
+satisfaction, emotional state, and business implications.
 
 Business Context: {business_context}
 
@@ -157,13 +157,15 @@ Analyze tone indicators like:
 - Urgency: "urgent", "immediately", "ASAP", "critical"
 - Threat: "cancel", "switch", "competitor", "lawyer", "report"
 
-IMPORTANT: Provide analysis responses in the same language as the input text, except for the predefined categorical fields (sentiment, satisfaction_level, emotional_state, churn_risk, relationship_status, response_approach) which must use the exact English values specified above. For example, if the input is in Spanish, provide tone_indicators in Spanish, but use English values like "positive" for sentiment.
+IMPORTANT: Provide analysis responses in the same language as the input text, except for the
+predefined categorical fields (sentiment, satisfaction_level, emotional_state, churn_risk,
+relationship_status, response_approach) which must use the exact English values specified above.
+For example, if the input is in Spanish, provide tone_indicators in Spanish, but use English
+values like "positive" for sentiment.
 
 Provide comprehensive sentiment analysis with business context and recommended response strategy."""
 
-    return PreparedTask(
-        instructions=instructions, response_format=CustomerSentiment, temperature=temperature, top_p=top_p
-    )
+    return PreparedTask(instructions=instructions, response_format=CustomerSentiment)
 
 
 # Backward compatibility - default configuration

openaivec/task/customer_support/inquiry_classification.py

@@ -8,7 +8,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -85,15 +85,18 @@ Example:
 
     # Extract classification components
     extracted_df = df.ai.extract("classification")
-    print(extracted_df[["inquiry", "classification_category", "classification_subcategory", "classification_confidence"]])
+    print(extracted_df[[
+        "inquiry", "classification_category",
+        "classification_subcategory", "classification_confidence"
+    ]])
     ```
 """
 
-from typing import Dict, List, Literal, Optional
+from typing import Dict, Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["inquiry_classification"]
 
@@ -103,7 +106,7 @@ class InquiryClassification(BaseModel):
     subcategory: str = Field(description="Specific subcategory within the primary category")
     confidence: float = Field(description="Confidence score for classification (0.0-1.0)")
     routing: str = Field(description="Recommended routing destination")
-    keywords: List[str] = Field(description="Key terms that influenced the classification")
+    keywords: list[str] = Field(description="Key terms that influenced the classification")
     priority: Literal["low", "medium", "high", "urgent"] = Field(
         description="Suggested priority level (low, medium, high, urgent)"
     )
@@ -111,27 +114,23 @@ class InquiryClassification(BaseModel):
 
 
 def inquiry_classification(
-    categories: Optional[Dict[str, List[str]]] = None,
-    routing_rules: Optional[Dict[str, str]] = None,
-    priority_rules: Optional[Dict[str, str]] = None,
+    categories: Dict[str, list[str]] | None = None,
+    routing_rules: Dict[str, str] | None = None,
+    priority_rules: Dict[str, str] | None = None,
     business_context: str = "general customer support",
-    custom_keywords: Optional[Dict[str, List[str]]] = None,
-    temperature: float = 0.0,
-    top_p: float = 1.0,
+    custom_keywords: Dict[str, list[str]] | None = None,
 ) -> PreparedTask:
     """Create a configurable inquiry classification task.
 
     Args:
-        categories (Optional[Dict[str, List[str]]]): Dictionary mapping category names to lists of subcategories.
+        categories (dict[str, list[str]] | None): Dictionary mapping category names to lists of subcategories.
             Default provides standard support categories.
-        routing_rules (Optional[Dict[str, str]]): Dictionary mapping categories to routing destinations.
+        routing_rules (dict[str, str] | None): Dictionary mapping categories to routing destinations.
             Default provides standard routing options.
-        priority_rules (Optional[Dict[str, str]]): Dictionary mapping keywords/patterns to priority levels.
+        priority_rules (dict[str, str] | None): Dictionary mapping keywords/patterns to priority levels.
             Default uses standard priority indicators.
         business_context (str): Description of the business context to help with classification.
-        custom_keywords (Optional[Dict[str, List[str]]]): Dictionary mapping categories to relevant keywords.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
+        custom_keywords (dict[str, list[str]] | None): Dictionary mapping categories to relevant keywords.
 
     Returns:
         PreparedTask configured for inquiry classification.
@@ -214,7 +213,8 @@ def inquiry_classification(
         for category, keywords in custom_keywords.items():
             keywords_text += f"- {category}: {', '.join(keywords)}\n"
 
-    instructions = f"""Classify the customer inquiry into the appropriate category and subcategory based on the configured categories and business context.
+    instructions = f"""Classify the customer inquiry into the appropriate category and subcategory
+based on the configured categories and business context.
 
 Business Context: {business_context}
 
@@ -243,13 +243,14 @@ Consider:
 - Business impact
 - Customer type indicators
 
-IMPORTANT: Provide analysis responses in the same language as the input text, except for the predefined categorical fields (priority) which must use the exact English values specified above. Category, subcategory, routing, and keywords should reflect the content and can be in the input language where appropriate, but priority must use English values like "high".
+IMPORTANT: Provide analysis responses in the same language as the input text, except for the
+predefined categorical fields (priority) which must use the exact English values specified above.
+Category, subcategory, routing, and keywords should reflect the content and can be in the input
+language where appropriate, but priority must use English values like "high".
 
 Provide accurate classification with detailed reasoning."""
 
-    return PreparedTask(
-        instructions=instructions, response_format=InquiryClassification, temperature=temperature, top_p=top_p
-    )
+    return PreparedTask(instructions=instructions, response_format=InquiryClassification)
 
 
 # Backward compatibility - default configuration

openaivec/task/customer_support/inquiry_summary.py

@@ -9,7 +9,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -59,11 +59,11 @@ Attributes:
         top_p=1.0 for deterministic output.
 """
 
-from typing import List, Literal
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["inquiry_summary"]
 
@@ -71,15 +71,15 @@ __all__ = ["inquiry_summary"]
 class InquirySummary(BaseModel):
     summary: str = Field(description="Concise summary of the customer inquiry (2-3 sentences)")
     main_issue: str = Field(description="Primary problem or request being addressed")
-    secondary_issues: List[str] = Field(description="Additional issues mentioned in the inquiry")
+    secondary_issues: list[str] = Field(description="Additional issues mentioned in the inquiry")
     customer_background: str = Field(description="Relevant customer context or history mentioned")
-    actions_taken: List[str] = Field(description="Steps the customer has already attempted")
+    actions_taken: list[str] = Field(description="Steps the customer has already attempted")
     timeline: str = Field(description="Timeline of events or when the issue started")
     impact_description: str = Field(description="How the issue affects the customer")
     resolution_status: Literal["not_started", "in_progress", "needs_escalation", "resolved"] = Field(
         description="Current status (not_started, in_progress, needs_escalation, resolved)"
     )
-    key_details: List[str] = Field(description="Important technical details, error messages, or specifics")
+    key_details: list[str] = Field(description="Important technical details, error messages, or specifics")
     follow_up_needed: bool = Field(description="Whether follow-up communication is required")
     summary_confidence: float = Field(description="Confidence in summary accuracy (0.0-1.0)")
 
@@ -87,16 +87,12 @@ class InquirySummary(BaseModel):
 def inquiry_summary(
     summary_length: str = "concise",
     business_context: str = "general customer support",
-    temperature: float = 0.0,
-    top_p: float = 1.0,
 ) -> PreparedTask:
     """Create a configurable inquiry summary task.
 
     Args:
         summary_length (str): Length of summary (concise, detailed, bullet_points).
         business_context (str): Business context for summary.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
 
     Returns:
         PreparedTask configured for inquiry summarization.
@@ -108,7 +104,8 @@ def inquiry_summary(
         "bullet_points": "Create a bullet-point summary with key facts and actions",
     }
 
-    instructions = f"""Create a comprehensive summary of the customer inquiry that captures all essential information for support agents and management.
+    instructions = f"""Create a comprehensive summary of the customer inquiry that captures all
+essential information for support agents and management.
 
 Business Context: {business_context}
 Summary Style: {length_instructions.get(summary_length, length_instructions["concise"])}
@@ -154,11 +151,15 @@ Focus on:
 - Clear distinction between symptoms and root causes
 - Relevant background without unnecessary details
 
-IMPORTANT: Provide summary responses in the same language as the input text, except for the predefined categorical field (resolution_status) which must use the exact English values specified above (not_started, in_progress, needs_escalation, resolved). For example, if the input is in German, provide all summary content in German, but use English values like "in_progress" for resolution_status.
+IMPORTANT: Provide summary responses in the same language as the input text, except for the
+predefined categorical field (resolution_status) which must use the exact English values
+specified above (not_started, in_progress, needs_escalation, resolved). For example, if the
+input is in German, provide all summary content in German, but use English values like
+"in_progress" for resolution_status.
 
 Provide accurate, actionable summary that enables efficient support resolution."""
 
-    return PreparedTask(instructions=instructions, response_format=InquirySummary, temperature=temperature, top_p=top_p)
+    return PreparedTask(instructions=instructions, response_format=InquirySummary)
 
 
 # Backward compatibility - default configuration

openaivec/task/customer_support/intent_analysis.py

@@ -8,7 +8,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -52,16 +52,16 @@ Example:
     ```
 
 Attributes:
-    INTENT_ANALYSIS (PreparedTask): A prepared task instance
-        configured for intent analysis with temperature=0.0 and
-        top_p=1.0 for deterministic output.
+    INTENT_ANALYSIS (PreparedTask): A prepared task instance configured for intent
+        analysis. Provide ``temperature=0.0`` and ``top_p=1.0`` to your API calls
+        for deterministic output.
 """
 
-from typing import List, Literal
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["intent_analysis"]
 
@@ -77,36 +77,34 @@ class IntentAnalysis(BaseModel):
         "request_feature",
         "provide_feedback",
     ] = Field(
-        description="Primary customer intent (get_help, make_purchase, cancel_service, get_refund, report_issue, seek_information, request_feature, provide_feedback)"
+        description="Primary customer intent (get_help, make_purchase, cancel_service, "
+        "get_refund, report_issue, seek_information, request_feature, provide_feedback)"
     )
-    secondary_intents: List[str] = Field(description="Additional intents if multiple goals are present")
+    secondary_intents: list[str] = Field(description="Additional intents if multiple goals are present")
     action_required: Literal[
         "provide_information", "troubleshoot", "process_request", "escalate", "redirect", "schedule_callback"
     ] = Field(
-        description="Required action (provide_information, troubleshoot, process_request, escalate, redirect, schedule_callback)"
+        description="Required action (provide_information, troubleshoot, process_request, "
+        "escalate, redirect, schedule_callback)"
     )
     intent_confidence: float = Field(description="Confidence in intent detection (0.0-1.0)")
     success_likelihood: Literal["very_high", "high", "medium", "low", "very_low"] = Field(
         description="Likelihood of successful resolution (very_high, high, medium, low, very_low)"
     )
     customer_goal: str = Field(description="What the customer ultimately wants to achieve")
-    implicit_needs: List[str] = Field(description="Unstated needs or concerns that may need addressing")
-    blocking_factors: List[str] = Field(description="Potential obstacles to achieving customer goal")
-    next_steps: List[str] = Field(description="Recommended next steps to address customer intent")
+    implicit_needs: list[str] = Field(description="Unstated needs or concerns that may need addressing")
+    blocking_factors: list[str] = Field(description="Potential obstacles to achieving customer goal")
+    next_steps: list[str] = Field(description="Recommended next steps to address customer intent")
     resolution_complexity: Literal["simple", "moderate", "complex", "very_complex"] = Field(
         description="Complexity of resolution (simple, moderate, complex, very_complex)"
     )
 
 
-def intent_analysis(
-    business_context: str = "general customer support", temperature: float = 0.0, top_p: float = 1.0
-) -> PreparedTask:
+def intent_analysis(business_context: str = "general customer support") -> PreparedTask:
     """Create a configurable intent analysis task.
 
     Args:
         business_context (str): Business context for intent analysis.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
 
     Returns:
         PreparedTask configured for intent analysis.
@@ -161,11 +159,15 @@ Pay attention to:
 - Urgency indicators: Time pressure affects resolution approach
 - Previous interactions: References to prior support contacts
 
-IMPORTANT: Provide analysis responses in the same language as the input text, except for the predefined categorical fields (primary_intent, action_required, success_likelihood, resolution_complexity) which must use the exact English values specified above. For example, if the input is in Japanese, provide customer_goal, implicit_needs, blocking_factors, next_steps, and reasoning in Japanese, but use English values like "get_help" for primary_intent.
+IMPORTANT: Provide analysis responses in the same language as the input text, except for the
+predefined categorical fields (primary_intent, action_required, success_likelihood,
+resolution_complexity) which must use the exact English values specified above. For example,
+if the input is in Japanese, provide customer_goal, implicit_needs, blocking_factors,
+next_steps, and reasoning in Japanese, but use English values like "get_help" for primary_intent.
 
 Provide comprehensive intent analysis with actionable recommendations."""
 
-    return PreparedTask(instructions=instructions, response_format=IntentAnalysis, temperature=temperature, top_p=top_p)
+    return PreparedTask(instructions=instructions, response_format=IntentAnalysis)
 
 
 # Backward compatibility - default configuration

openaivec/task/customer_support/response_suggestion.py

@@ -9,7 +9,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -57,11 +57,11 @@ Attributes:
         top_p=1.0 for deterministic output.
 """
 
-from typing import List, Literal
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["response_suggestion"]
 
@@ -77,14 +77,14 @@ class ResponseSuggestion(BaseModel):
     response_type: Literal["acknowledgment", "solution", "escalation", "information_request", "closure"] = Field(
         description="Type of response (acknowledgment, solution, escalation, information_request, closure)"
     )
-    key_points: List[str] = Field(description="Main points that must be addressed in the response")
+    key_points: list[str] = Field(description="Main points that must be addressed in the response")
     follow_up_required: bool = Field(description="Whether follow-up communication is needed")
     escalation_suggested: bool = Field(description="Whether escalation to management is recommended")
-    resources_needed: List[str] = Field(description="Additional resources or information required")
+    resources_needed: list[str] = Field(description="Additional resources or information required")
     estimated_resolution_time: Literal["immediate", "hours", "days", "weeks"] = Field(
         description="Estimated time to resolution (immediate, hours, days, weeks)"
     )
-    alternative_responses: List[str] = Field(description="Alternative response options for different scenarios")
+    alternative_responses: list[str] = Field(description="Alternative response options for different scenarios")
     personalization_notes: str = Field(description="Suggestions for personalizing the response")
 
 
@@ -92,8 +92,6 @@ def response_suggestion(
     response_style: str = "professional",
     company_name: str = "our company",
     business_context: str = "general customer support",
-    temperature: float = 0.0,
-    top_p: float = 1.0,
 ) -> PreparedTask:
     """Create a configurable response suggestion task.
 
@@ -101,8 +99,6 @@ def response_suggestion(
         response_style (str): Style of response (professional, friendly, empathetic, formal).
         company_name (str): Name of the company for personalization.
         business_context (str): Business context for responses.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
 
     Returns:
         PreparedTask configured for response suggestions.
@@ -115,7 +111,8 @@ def response_suggestion(
         "formal": "Use formal business language appropriate for official communications",
     }
 
-    instructions = f"""Generate a professional, helpful response suggestion for the customer inquiry that addresses their needs effectively.
+    instructions = f"""Generate a professional, helpful response suggestion for the customer
+inquiry that addresses their needs effectively.
 
 Business Context: {business_context}
 Company Name: {company_name}
@@ -180,13 +177,16 @@ Avoid:
 - Dismissing customer concerns
 - Lengthy responses that don't address the main issue
 
-IMPORTANT: Generate responses in the same language as the input text, except for the predefined categorical fields (tone, priority, response_type, estimated_resolution_time) which must use the exact English values specified above. For example, if the input is in Italian, provide suggested_response, key_points, alternative_responses, and personalization_notes in Italian, but use English values like "empathetic" for tone.
+IMPORTANT: Generate responses in the same language as the input text, except for the predefined
+categorical fields (tone, priority, response_type, estimated_resolution_time) which must use
+the exact English values specified above. For example, if the input is in Italian, provide
+suggested_response, key_points, alternative_responses, and personalization_notes in Italian,
+but use English values like "empathetic" for tone.
 
-Generate helpful, professional response that moves toward resolution while maintaining positive customer relationship."""
+Generate helpful, professional response that moves toward resolution while maintaining
+positive customer relationship."""
 
-    return PreparedTask(
-        instructions=instructions, response_format=ResponseSuggestion, temperature=temperature, top_p=top_p
-    )
+    return PreparedTask(instructions=instructions, response_format=ResponseSuggestion)
 
 
 # Backward compatibility - default configuration

openaivec/task/customer_support/urgency_analysis.py

@@ -8,7 +8,7 @@ Example:
 
     ```python
     from openai import OpenAI
-    from openaivec.responses import BatchResponses
+    from openaivec import BatchResponses
     from openaivec.task import customer_support
 
     client = OpenAI()
@@ -96,11 +96,11 @@ Example:
     ```
 """
 
-from typing import Dict, List, Literal, Optional
+from typing import Dict, Literal
 
 from pydantic import BaseModel, Field
 
-from ...model import PreparedTask
+from openaivec._model import PreparedTask
 
 __all__ = ["urgency_analysis"]
 
@@ -111,10 +111,11 @@ class UrgencyAnalysis(BaseModel):
     )
     urgency_score: float = Field(description="Urgency score from 0.0 (not urgent) to 1.0 (extremely urgent)")
     response_time: Literal["immediate", "within_1_hour", "within_4_hours", "within_24_hours"] = Field(
-        description="Recommended response time from configured times (immediate, within_1_hour, within_4_hours, within_24_hours)"
+        description="Recommended response time from configured times "
+        "(immediate, within_1_hour, within_4_hours, within_24_hours)"
     )
     escalation_required: bool = Field(description="Whether this inquiry requires escalation to management")
-    urgency_indicators: List[str] = Field(description="Specific words or phrases that indicate urgency")
+    urgency_indicators: list[str] = Field(description="Specific words or phrases that indicate urgency")
     business_impact: Literal["none", "low", "medium", "high", "critical"] = Field(
         description="Potential business impact (none, low, medium, high, critical)"
     )
@@ -126,30 +127,26 @@ class UrgencyAnalysis(BaseModel):
 
 
 def urgency_analysis(
-    urgency_levels: Optional[Dict[str, str]] = None,
-    response_times: Optional[Dict[str, str]] = None,
-    customer_tiers: Optional[Dict[str, str]] = None,
-    escalation_rules: Optional[Dict[str, str]] = None,
-    urgency_keywords: Optional[Dict[str, List[str]]] = None,
+    urgency_levels: Dict[str, str] | None = None,
+    response_times: Dict[str, str] | None = None,
+    customer_tiers: Dict[str, str] | None = None,
+    escalation_rules: Dict[str, str] | None = None,
+    urgency_keywords: Dict[str, list[str]] | None = None,
     business_context: str = "general customer support",
     business_hours: str = "24/7 support",
-    sla_rules: Optional[Dict[str, str]] = None,
-    temperature: float = 0.0,
-    top_p: float = 1.0,
+    sla_rules: Dict[str, str] | None = None,
 ) -> PreparedTask:
     """Create a configurable urgency analysis task.
 
     Args:
-        urgency_levels (Optional[Dict[str, str]]): Dictionary mapping urgency levels to descriptions.
-        response_times (Optional[Dict[str, str]]): Dictionary mapping urgency levels to response times.
-        customer_tiers (Optional[Dict[str, str]]): Dictionary mapping tier names to descriptions.
-        escalation_rules (Optional[Dict[str, str]]): Dictionary mapping conditions to escalation actions.
-        urgency_keywords (Optional[Dict[str, List[str]]]): Dictionary mapping urgency levels to indicator keywords.
+        urgency_levels (dict[str, str] | None): Dictionary mapping urgency levels to descriptions.
+        response_times (dict[str, str] | None): Dictionary mapping urgency levels to response times.
+        customer_tiers (dict[str, str] | None): Dictionary mapping tier names to descriptions.
+        escalation_rules (dict[str, str] | None): Dictionary mapping conditions to escalation actions.
+        urgency_keywords (dict[str, list[str]] | None): Dictionary mapping urgency levels to indicator keywords.
         business_context (str): Description of the business context.
         business_hours (str): Description of business hours for response time calculation.
-        sla_rules (Optional[Dict[str, str]]): Dictionary mapping customer tiers to SLA requirements.
-        temperature (float): Sampling temperature (0.0-1.0).
-        top_p (float): Nucleus sampling parameter (0.0-1.0).
+        sla_rules (dict[str, str] | None): Dictionary mapping customer tiers to SLA requirements.
 
     Returns:
         PreparedTask configured for urgency analysis.
@@ -278,13 +275,15 @@ Consider:
 - Revenue or operational impact
 - Compliance or legal implications
 
-IMPORTANT: Provide analysis responses in the same language as the input text, except for the predefined categorical fields (urgency_level, response_time, business_impact, customer_tier) which must use the exact English values specified above. For example, if the input is in French, provide urgency_indicators and reasoning in French, but use English values like "critical" for urgency_level.
+IMPORTANT: Provide analysis responses in the same language as the input text, except for the
+predefined categorical fields (urgency_level, response_time, business_impact, customer_tier)
+which must use the exact English values specified above. For example, if the input is in French,
+provide urgency_indicators and reasoning in French, but use English values like "critical" for
+urgency_level.
 
 Provide detailed analysis with clear reasoning for urgency level and response time recommendations."""
 
-    return PreparedTask(
-        instructions=instructions, response_format=UrgencyAnalysis, temperature=temperature, top_p=top_p
-    )
+    return PreparedTask(instructions=instructions, response_format=UrgencyAnalysis)
 
 
 # Backward compatibility - default configuration