ޠj%<SrSSKJr SSKrSSKJr SSKJrJrJ r J r SSK J r J r Jr SSKJrJrJrJr SSKJrJr SS KJrJr \(aSS KJr \ S rS r"S S\5r"SS\ \ \\45r SS/r!g)z>Agent middleware that integrates OpenAI's moderation endpoint.) annotationsN)Sequence) TYPE_CHECKINGAnyLiteralcast)AgentMiddleware AgentState hook_config) AIMessage BaseMessage HumanMessage ToolMessage) AsyncOpenAIOpenAI) ModerationModerationModel)Runtime)inputoutputtoolzQI'm sorry, but I can't comply with that request. It was flagged for {categories}.cD^\rSrSrSrSU4SjjrSrU=r$)OpenAIModerationErrorzIRaised when OpenAI flags content and `exit_behavior` is set to `"error"`.cH>[TU]U5 XlX lX0lg)zInitialize the error with violation details. Args: content: The content that was flagged. stage: The stage where the violation occurred. result: The moderation result from OpenAI. message: The error message. N)super__init__contentstageresult)selfrrr message __class__s g/var/www/html/gaurav/venv/lib/python3.13/site-packages/langchain_openai/middleware/openai_moderation.pyrOpenAIModerationError.__init__s! !   )rr r) rstrrViolationStager rr"r'returnNone)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__ __classcell__r#s@r$rrsBS      r&rc ^\rSrSrSrSSSSSSSSS.S U4S jjjr\"S/S 9S!S j5r\"S/S 9S!S j5r\"S/S 9S!S j5r \"S/S 9S!Sj5r S"Sjr S"Sjr S"Sjr S"SjrS"SjrS"SjrS"SjrS"SjrS#SjrS$SjrS$SjrS%SjrS&SjrS'SjrS(SjrS)SjrSrU=r$)*OpenAIModerationMiddleware1z:Moderate agent traffic using OpenAI's moderation endpoint.zomni-moderation-latestTFendN)model check_input check_outputcheck_tool_results exit_behaviorviolation_messageclient async_clientc>[T U]5 XlX lX0lX@lXPlX`lXplXl g)aCreate the middleware instance. Args: model: OpenAI moderation model to use. check_input: Whether to check user input messages. check_output: Whether to check model output messages. check_tool_results: Whether to check tool result messages. exit_behavior: How to handle violations (`'error'`, `'end'`, or `'replace'`). violation_message: Custom template for violation messages. client: Optional pre-configured OpenAI client to reuse. If not provided, a new client will be created. async_client: Optional pre-configured AsyncOpenAI client to reuse. If not provided, a new async client will be created. N) rrr7r8r9r:r;r<_client _async_client) r!r7r8r9r:r;r<r=r>r#s r$r#OpenAIModerationMiddleware.__init__4s>6  &("4*!2 )r&) can_jump_tocUR(dUR(dg[URS/55nU(dgUR U5$)zModerate user input and tool results before the model is called. Args: state: Current agent state containing messages. runtime: Agent runtime context. Returns: Updated state with moderated messages, or `None` if no changes. Nmessages)r8r:listget_moderate_inputsr!stateruntimerEs r$ before_model'OpenAIModerationMiddleware.before_modelZsD(?(? *b12$$X..r&cUR(dg[URS/55nU(dgURU5$)zModerate model output after the model is called. Args: state: Current agent state containing messages. runtime: Agent runtime context. Returns: Updated state with moderated messages, or `None` if no changes. NrE)r9rFrG_moderate_outputrIs r$ after_model&OpenAIModerationMiddleware.after_modelps<   *b12$$X..r&c# UR(dUR(dg[URS/55nU(dgUR U5IShvN $N7f)zAsync version of before_model. Args: state: Current agent state containing messages. runtime: Agent runtime context. Returns: Updated state with moderated messages, or `None` if no changes. NrE)r8r:rFrG_amoderate_inputsrIs r$ abefore_model(OpenAIModerationMiddleware.abefore_modelsM(?(? *b12++H5555sAA$A"A$c# UR(dg[URS/55nU(dgURU5IShvN $N7f)zAsync version of after_model. Args: state: Current agent state containing messages. runtime: Agent runtime context. Returns: Updated state with moderated messages, or `None` if no changes. NrE)r9rFrG_amoderate_outputrIs r$ aafter_model'OpenAIModerationMiddleware.aafter_modelsE   *b12++H5555sA A A Ac<[U5nSnUR(a1URU5nU(aSU;aU$[SUS5nSnUR(a1UR U5nU(aSU;aU$[SUS5nSnU(aSU0$gNFjump_tolist[BaseMessage]rET)rFr:_moderate_tool_messagesrr8_moderate_user_messager!rEworkingmodifiedactions r$rH+OpenAIModerationMiddleware._moderate_inputssx.  " "11':F&!M2F:4FG   009F&!M2F:4FG ( (r&cl# [U5nSnUR(a9URU5IShvN nU(aSU;aU$[SUS5nSnUR(a9UR U5IShvN nU(aSU;aU$[SUS5nSnU(aSU0$gNzN27fr[)rFr:_amoderate_tool_messagesrr8_amoderate_user_messager`s r$rS,OpenAIModerationMiddleware._amoderate_inputssx.  " "88AAF&!M2F:4FG   77@@F&!M2F:4FG ( (%BAs"3B4B0A B4?B21B42B4cURU[5nUcgXnURU5nU(dgURU5nUR(dgUR XSXES9$Nrindexrrr )_find_last_indexr _extract_text _moderateflagged_apply_violationr!rE last_ai_idx ai_messagetextr s r$rO+OpenAIModerationMiddleware._moderate_outputst++Hi@  * !!*-%~~$$ x%  r&c# URU[5nUcgXnURU5nU(dgURU5IShvN nUR(dgUR XSXES9$N'7frj)rmr rn _amoderaterprqrrs r$rW,OpenAIModerationMiddleware._amoderate_outputs++Hi@  * !!*-t,,~~$$ x%  -A A8A6(A8c URU[5nUcg[U5nSn[US-[ U55HnX5n[ U[ 5(dMURU5nU(dM8URU5nUR(dM\URX5SXxS9n U (dMvSU ;aU s $[SU S5nSnM U(aSU0$g NFrrkr\r]rET) rmr rFrangelen isinstancerrnrorprqr r!rErsrarbidxmsgrur rcs r$r^2OpenAIModerationMiddleware._moderate_tool_messagess++Hi@  x.q#g,7C,Cc;//%%c*D^^D)F>>**&$+Fv&!M2F:4FG)8, ( (r&c # URU[5nUcg[U5nSn[US-[ U55HnX5n[ U[ 5(dMURU5nU(dM8URU5IShvN nUR(dMdURX5SXxS9n U (dM~SU ;aU s $[SU S5nSnM U(aSU0$gN\7fr|) rmr rFr~rrrrnrxrprqrrs r$rf3OpenAIModerationMiddleware._amoderate_tool_messages5s++Hi@  x.q#g,7C,Cc;//%%c*D??400F>>**&$+Fv&!M2F:4FG)8, ( (!1sB C-C+-C-,C-cURU[5nUcgXnURU5nU(dgURU5nUR(dgUR XSXES9$Nrrk)rmrrnrorprqr!rErr"rur s r$r_1OpenAIModerationMiddleware._moderate_user_messageZsr##Hl; ;-!!'*%~~$$ w%  r&c# URU[5nUcgXnURU5nU(dgURU5IShvN nUR(dgUR XSXES9$N'7fr)rmrrnrxrprqrs r$rg2OpenAIModerationMiddleware._amoderate_user_messagens}##Hl; ;-!!'*t,,~~$$ w%  -rzcURXE5nURS:Xa [UUUUS9eURS:Xa S[US9/S.$Ucg[ U5nXrn[ [ URSU0S95Xr'SU0$) Nerror)rrr r"r6)r)r\rEr)updaterE)_format_violation_messager;rr rFrr model_copy) r!rErlrrr violation_text new_messagesoriginals r$rq+OpenAIModerationMiddleware._apply_violations77H    ('&      &$9^3T2UV V =H~ &" ,,Y4O,P L))r&cURcUR5UlURRRURUS9nUR S$N)r7rr)r@ _build_client moderationscreater7resultsr!ruresponses r$ro$OpenAIModerationMiddleware._moderatesP << --/DL<<++2242P""r&c# URcUR5UlURRRURUS9IShvN nUR S$N7fr)rA_build_async_clientrrr7rrs r$rx%OpenAIModerationMiddleware._amoderatesi    %!%!9!9!;D ++77>>**D?  "" sAA,A*A,c8[5UlUR$N)rr@r!s r$r(OpenAIModerationMiddleware._build_clientsx ||r&c8[5UlUR$r)rrArs r$r.OpenAIModerationMiddleware._build_async_clients(]!!!r&cURR5nUR5VVs/sH upEU(dMURSS5PM" nnnU(aSR U5OSnUR =(d [ n[R"URR5SS9n URUU US9n U $s snnf![a Un U $f=f)N_ z, zOpenAI's safety policiesT) sort_keys) categoriescategory_scoresoriginal_content) r model_dumpitemsreplacejoinr<DEFAULT_VIOLATION_TEMPLATEjsondumpsrformatKeyError) r!rr categories_dictnamerprcategory_labeltemplate scores_jsonr"s r$r4OpenAIModerationMiddleware._format_violation_messages ++668"1!6!6!8 !8  #DLLc "!8  &0DIIj !5O ))G-Gjj!7!7!B!B!DPTU  oo) +!(&G%  G s C C 8C C! C!cr[[U5S- SS5Hn[XU5(dMUs $ g)Nr})r~rr)r!rE message_typers r$rm+OpenAIModerationMiddleware._find_last_indexs9X*B3C(-66 4r&cURcg[USS5nUc[UR5$[U5nU(aU$S$)Nru)rgetattrr')r!r" text_accessorrus r$rn(OpenAIModerationMiddleware._extract_textsI ?? "6  w' '=!t%%r&)rAr@r8r9r:r;r7r<)r7rr8boolr9rr:rr;z"Literal['error', 'end', 'replace']r< str | Noner=z OpenAI | Noner>zAsyncOpenAI | Noner)r*)rJzAgentState[Any]rKz Runtime[Any]r)dict[str, Any] | None)rESequence[BaseMessage]r)r) rErrl int | Nonerr(rr'r rr)r)rur'r)r)r)r)r)r)rr'r rr)r')rErrztype[BaseMessage]r)r)r"r r)r)r+r,r-r.r/rr rLrPrTrXrHrSrOrWr^rfr_rgrqrorxrrrrmrnr0r1r2s@r$r4r41s{D ": !#(rs}D" $44VVUU&4)23X L2m&C#1E!Fm&b   r&