h|SrSSKrSSKrSSKJr SSKJrJrJr SSKJ r SSK J r SSK J r SSKJr \R "\5r"S S 5rg) z5 LangSmith integration handler for tracking AI usage N)datetime)OptionalDictAny)Client)LangChainTracer)capture_exception) MongoHandlerc\rSrSrSrS\S\\4SjrS S\S\\S\ 4S jjr S\S \ \\ \\44S\ \\ 44S jr S rg)LangSmithHandlerc[5Ul[5Ul[ 5Ul[ RS5 g![a2n[ RS[U535 [U5 eSnAff=f)zInitialize LangSmith handlerz*Successfully initialized LangSmith handlerz&Error initializing LangSmith handler: N) rclientrtracerr mongo_handlerloggerinfo Exceptionerrorstrr )selfes ;D:\Projects\AI-TestCaseGenerator\utils\langsmith_handler.py__init__LangSmithHandler.__init__sf !(DK)+DK!-D  KKD E  LLA#a&J K a  sAA B-A<<Brun_idreturncJURRU5nURRS05nURS5nURS5nU(aU(d[R SU35 gUUUUR =(d SUR (aUR RS5SOSUR=(d SUR=(d SUR=(d [R"5URURUR(a UROS UR UR(a[#UR5OSUR$UR&URS S5S .S . nUR(R+U5n[R-S U35 U$![.a9n[RSUS[#U535 [1U5 SnAgSnAff=f)z Process a LangSmith run and store usage data. User information is extracted from run metadata. Args: run_id (str): LangSmith run ID Returns: Optional[str]: MongoDB record ID if successful, None otherwise metadatauser_id user_emailz!No user information found in run Nunknown/rSuccesssource)tagsroutputsfeedbackr%) rr r! model_name ai_provider input_tokens output_tokens timestampend_time latency_msstatusrz&Successfully tracked AI usage for run zError processing LangSmith run : )rread_runextragetrwarningmodelsplit prompt_tokenscompletion_tokens start_timerutcnowr. runtime_msrr&rr'r(rtrack_ai_usagerrr ) rrrunrr r! usage_data record_idrs r process_runLangSmithHandler.process_runs+ ++&&v.Cyy}}Z4Hll9-G!l3J*!B6(KL!"(!ii49:=))syys3A6 # 1 1 6Q!$!6!6!;! ^^@x/@LL!nn'*yy#))iHH/2yyS^d"{{ # &ll8Y? J,**99*EI KK@I J   LL:6("SVHM N a  sA?GEG H")/HH"N project_namer:c URRUUS9n[U5$![a;n[R SUS[ U535 [U5 /sSnA$SnAff=f)z Get all runs for a specific project Args: project_name (str): Name of the LangSmith project start_time (datetime, optional): Only get runs after this time Returns: list: List of run data )rCr:zError getting runs for project r1N)r list_runslistrrrrr )rrCr:runsrs rget_project_runs!LangSmithHandler.get_project_runsUsm ;;(()%)D:   LL:<.3q6(S T a I s$' A,0A'!A,'A, user_mappingc URU5nSnSnSnUH|nSnUR5H!upXR=(d /;dMU n O U(dUS- nMHURURUSUS5(aUS- nMwUS- nM~ [ U5UUUS.$![ aIn [RSUS[U 535 [U 5 [U 5SSSS .sSn A $Sn A ff=f) aU Sync all runs from a project to our tracking system Args: project_name (str): Name of the LangSmith project user_mapping (dict): Mapping of run identifiers to user data Example: { "session_id": {"user_id": "123", "user_email": "user@example.com"} } Returns: dict: Summary of sync operation rNr r!) total_runs processedfailedskippedzError syncing project r1)rrNrOrP) rHitemsr&rAidlenrrrrr ) rrCrJrGrNrOrPr> user_data identifierdatars rsync_project_runs"LangSmithHandler.sync_project_runsks( ((6DIFG (4(:(:(<$J!hhn"5$( )= !qLG##CFFIi,@)LBYZZNIaKF!&"$i& "    LL1,r#a&J K a Q   s%A B+AB++ C>5>C93C>9C>)rrr)N)__name__ __module__ __qualname____firstlineno__rrrrArrFrHrrrW__static_attributes__rr r s{ 5#5(3-5nShx>P\`,6c6c4PSUXPX>FYAZ6_cdgildl_m6r_r )__doc__osloggingrtypingrrr langsmithr%langchain.callbacks.tracers.langchainrutils.error_loggerr utils.mongo_handlerr getLoggerrYrr r^r_rrisC &&A0,   8 $RRr_