
    g\                        d dl mZ d dlZd dlmZ d dlmZmZ d dl	m
Z
mZmZmZ ddlmZ ddlmZmZmZ e G d	 d
ee                Zy)    )annotationsN)	Awaitable)	dataclassfield)AnyCallableGenericcast   )_utils)	AgentDeps
RunContextSystemPromptFuncc                  l    e Zd ZU ded<   dZded<    ed      Zded<    ed      Zded<   d	 Zdd
Z	y)SystemPromptRunnerzSystemPromptFunc[AgentDeps]functionFbooldynamic)init
_takes_ctx	_is_asyncc                    t        t        j                  | j                        j                        dkD  | _        t        j                  | j                        | _        y )Nr   )leninspect	signaturer   
parametersr   iscoroutinefunctionr   )selfs    O/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/_system_prompt.py__post_init__z SystemPromptRunner.__post_init__   s>    g//>IIJQN 44T]]C    c                R  K   | j                   r|f}nd}| j                  r<t        t        t        gt
        t           f   | j                        } ||  d {   S t        t        t        gt        f   | j                        }t        j                  |g|  d {   S 7 J7 w)N )
r   r   r
   r   r   r   strr   r   run_in_executor)r   run_contextargsr   s       r   runzSystemPromptRunner.run   s     ??>DD>>HcUIcN%:;T]]KH!4((HcUCZ0$--@H//@4@@@ ) As%   AB'B#AB'B%B'%B'N)r&   zRunContext[AgentDeps]returnr$   )
__name__
__module____qualname____annotations__r   r   r   r   r    r(   r#   r!   r   r   r      s<    ))GT%(J('It'DAr!   r   )
__future__r   _annotationsr   collections.abcr   dataclassesr   r   typingr   r   r	   r
    r   toolsr   r   r   r   r#   r!   r   <module>r5      sC    2  % ( / /  : : A+ A Ar!   