jQ UdZddlmZddlZddlmZmZmZmZm Z m Z m Z ddl Z ejeZehdZded< d dZd d Zdd Z dd Zy)z9Adapt OpenAI-style messages to Anthropic API conventions.) annotationsN)AnyDictListOptionalSetTupleType> modeltoolstop_ktop_psystemmessagesmetadata max_tokens temperature tool_choicestop_sequenceszfrozenset[str]_SUPPORTED_PARAMScg}g}|D]I}|jddk(r"|j|jdd9|j|K|rdj|nd}||fS)zSeparate system messages from conversation messages. Anthropic requires system content as a top-level parameter rather than a message with role="system". rolercontentz N)getappendjoin)r system_parts non_systemmsg system_texts /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/evaluation/models/anthropic/message_adapter.pyextract_system_messagesr#sn!L')J 776?h &    2 6 7   c "  0<&++l+K  ""cT|j}|jdddd|diS)zBuild an Anthropic ``output_config`` dict from a pydantic model. Uses the native ``json_schema`` output format (SDK >=0.85) which constrains the model to produce JSON matching the schema directly, without the tool_use indirection. titleNformat json_schema)typeschema)model_json_schemapop)r r*s r"pydantic_to_output_configr-1s8 $ $ &F JJw!  r$cD|jdr|tddS|S)Nz anthropic/) startswithlen) model_names r"strip_anthropic_prefixr2Bs)\*#l+-.. r$ci}|jD]?\}}|tvr|||<||vstjd||j |A|S)Nz.Dropping unsupported Anthropic parameter '%s'.)itemsrLOGGERdebugadd)paramsalready_warnedfilteredkeyvalues r"filter_unsupported_paramsr=Hs` "Hlln U # #!HSM  & LL@    s #% Or$)rzList[Dict[str, Any]]returnz*Tuple[Optional[str], List[Dict[str, Any]]])r zType[pydantic.BaseModel]r>Dict[str, Any])r1strr>r@)r8r?r9zSet[str]r>r?)__doc__ __future__rloggingtypingrrrrrr r pydantic getLogger__name__r5 frozensetr__annotations__r#r-r2r=r$r"rKs?">>>   8 $%. %>"#"#/#&"  r$