.hddlZddlmZddlmZddlmZmZddlm Z m Z m Z ddl m Z mZddlmZdd lmZdd lmZmZd ZGd d ejZedGddeZdedefdZdeedefdZdS)N) dataclass)Any) BaseModel TypeAdapter) TypedDictget_args get_origin)ModelBehaviorError UserError)ensure_strict_json_schema) SpanError)_error_tracing_jsonresponseceZdZdZejdefdZejdefdZ ejde ee ffdZ ejdefdZ ejdede fdZd S) AgentOutputSchemaBaseAn object that captures the JSON schema of the output, as well as validating/parsing JSON produced by the LLM into the output type. returncdSz=Whether the output type is plain text (versus a JSON object).Nselfs P/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/agent_output.py is_plain_textz#AgentOutputSchemaBase.is_plain_text  cdS)The name of the output type.Nrrs rnamezAgentOutputSchemaBase.namerrcdS)zqReturns the JSON schema of the output. Will only be called if the output type is not plain text. Nrrs r json_schemaz!AgentOutputSchemaBase.json_schema rcdS)zWhether the JSON schema is in strict mode. Strict mode constrains the JSON schema features, but guarantees valis JSON. See here for details: https://platform.openai.com/docs/guides/structured-outputs#supported-schemas Nrrs ris_strict_json_schemaz+AgentOutputSchemaBase.is_strict_json_schema&s rjson_strcdS)zValidate a JSON string against the output type. You must return the validated object, or raise a `ModelBehaviorError` if the JSON is invalid. Nr)rr's r validate_jsonz#AgentOutputSchemaBase.validate_json.r$rN)__name__ __module__ __qualname____doc__abcabstractmethodboolrstrr!dictrr#r&r)rrrrrs  t      c      T#s(^      t      c c      rrF)initceZdZUdZeeed< eeed< eed< e e efed< eed< ddeedefd Z d efd Z d efd Z d e e effd Zde d efdZd e fdZdS)AgentOutputSchemar output_type _type_adapter _is_wrapped_output_schema_strict_json_schemaTstrict_json_schemach||_||_| |tur;d|_t ||_|j|_dSt| |_|jrJtdt|i}t ||_|j|_n2t ||_|j|_|jr> t|j|_dS#t$r}td|d}~wwxYwdS)a  Args: output_type: The type of the output. strict_json_schema: Whether the JSON schema is in strict mode. We **strongly** recommend setting this to True, as it increases the likelihood of correct JSON input. NF OutputTypezStrict JSON schema is enabled, but the output type is not valid. Either make the output type strict, or pass output_schema_strict=False to your Agent()) r6r:r1r8rr7r#r9"_is_subclass_of_base_model_or_dictr_WRAPPER_DICT_KEYr r )rr6r;r=es r__init__zAgentOutputSchema.__init__OsW'#5  +"4"4$D !,[!9!9D "&"4"@"@"B"BD  F B+NNN   C"%{J "-Z!8!8D "&"4"@"@"B"BD  !,[!9!9D "&"4"@"@"B"BD   #  &?@S&T&T###   #    s5D D/D**D/rc0|jdup |jtuSr)r6r1rs rrzAgentOutputSchema.is_plain_textzs4'B4+;s+BBrc|jS)z*Whether the JSON schema is in strict mode.)r:rs rr&z'AgentOutputSchema.is_strict_json_schema~s ''rcV|rtd|jS)z#The JSON schema of the output type.z9Output type is plain text, so no JSON schema is available)rr r9rs rr#zAgentOutputSchema.json_schemas/      YWXX X""rr'c tj||jd}|jrt |t sWt jtdddt|itdt|d|t|vrHt jtdddtd itdtd ||tS|S) zValidate a JSON string against the output type. Returns the validated object, or raises a `ModelBehaviorError` if the JSON is invalid. F)partialz Invalid JSONdetailszExpected a dict, got )messagedataz for JSON: zCould not find key z in JSONz in JSON: ) rr)r7r8 isinstancer2rattach_error_to_current_spanrtyper r?)rr' validateds rr)zAgentOutputSchema.validate_jsons1'$2DeTTT   0i.. ; .')Ri)R)RS )RDOORRRR! 11; .')Z?P)Z)Z)Z[ )Q*;QQxQQ./ /rc*t|jS)r ) _type_to_strr6rs rr!zAgentOutputSchema.namesD,---rN)T)r*r+r,r-rLr__annotations__rr0r2r1rArr&r#r)r!rrrr5r56sKc!s####RcN"""())DI)4))))VCtCCCC(t((((#T#s(^#### cc<.c......rr5trct|tsdSt|}ttf}t |p||S)NF)rJrLr rr2 issubclass)rQorigin allowed_typess rr>r>sD a  u]]F%M fk= 1 11rct|}t|}||jS|r,dd|D}|jd|dSt |S)Nz, c34K|]}t|VdS)N)rO).0args r z_type_to_str..s*??3\#..??????r[])r rr*joinr1)rQrTargsargs_strs rrOrOsq ]]F A;;D ~z 99??$?????///H////1vv r)r. dataclassesrtypingrpydanticrrtyping_extensionsrrr exceptionsr r strict_schemar tracingrutilrrr?ABCrr5r0r>rLr1rOrrrris !!!!!!++++++++==========55555555444444''''''''# # # # # CG# # # L q.q.q.q.q.-q.q.q.h 2# 2$ 2 2 2 2 DI #      r