.h(ddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z ddl mZmZdd lmZdd lmZGd d eZGd deZGddeZeZeeZddZddZdS)) annotationsN)cached_property)Any)logger)TracingExporterTracingProcessor)Span)TraceceZdZdZddZdS) ConsoleSpanExporterz+Prints the traces and spans to the console.itemslist[Trace | Span[Any]]returnNonec|D]\}t|tr!td|jd|jd8td|]dS)Nzxporter] Export trace_id=z, name=z, z[Exporter] Export span: ) isinstancer printtrace_idnameexport)selfritems V/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/tracing/processors.pyrzConsoleSpanExporter.exports} B BD$&& BWDMWW$)WWWXXXX@@@AAAA  B BNrrrr)__name__ __module__ __qualname____doc__rrrrrs455BBBBBBrrc~eZdZ dddZddZedZedZedZddZ dZ dS) BackendSpanExporterN'https://api.openai.com/v1/traces/ingest?>@api_key str | None organizationprojectendpointstr max_retriesint base_delayfloat max_delayc||_||_||_||_||_||_||_tjtj dd|_ dS)a Args: api_key: The API key for the "Authorization" header. Defaults to `os.environ["OPENAI_API_KEY"]` if not provided. organization: The OpenAI organization to use. Defaults to `os.environ["OPENAI_ORG_ID"]` if not provided. project: The OpenAI project to use. Defaults to `os.environ["OPENAI_PROJECT_ID"]` if not provided. endpoint: The HTTP endpoint to which traces/spans are posted. max_retries: Maximum number of retries upon failures. base_delay: Base delay (in seconds) for the first backoff. max_delay: Maximum delay (in seconds) for backoff growth. <@)timeoutconnectr7N) _api_key _organization_projectr-r/r1r3httpxClientTimeout_client)rr)r+r,r-r/r1r3s r__init__zBackendSpanExporter.__init__s`.  )   &$"|EM"c,R,R,RSSS rc"||_||_dS)zSet the OpenAI API key for the exporter. Args: api_key: The OpenAI API key to use. This is the same key used by the OpenAI Python client. N)r:r))rr)s r set_api_keyzBackendSpanExporter.set_api_keyAs   rcN|jptjdS)NOPENAI_API_KEY)r:osenvirongetrs rr)zBackendSpanExporter.api_keyLs}@ /? @ @@rcN|jptjdS)N OPENAI_ORG_ID)r;rFrGrHrIs rr+z BackendSpanExporter.organizationPs!DRZ^^O%D%DDrcN|jptjdS)NOPENAI_PROJECT_ID)r<rFrGrHrIs rr,zBackendSpanExporter.projectTs}C /B C CCrrrrrcx|sdS|jstjddSd|D}d|i}d|jddd}|jr |j|d<|jr |j|d <d }|j} |d z } |j|j|| }|j dkr'tj dt|ddSd|j cxkrdkr)nn&tj d|j d|j dStjd|j dn3#tj$r!}tjd|Yd}~nd}~wwxYw||jkrtj ddS|t#jd d|zz} t'j| t+|dz|j}P)Nz0OPENAI_API_KEY is not set, skipping trace exportc^g|]*}||+Sr")r).0rs r z.BackendSpanExporter.export..`s-AAA$4;;==A AAArdatazBearer zapplication/jsonz traces=v1) Authorizationz Content-Typez OpenAI-BetazOpenAI-OrganizationzOpenAI-ProjectrTr)urlheadersjsoni,z Exported z itemsiiz![non-fatal] Tracing client error z: z"[non-fatal] Tracing: server error z , retrying.z%[non-fatal] Tracing: request failed: zB[non-fatal] Tracing: max retries reached, giving up on this batch.g?r)r)rwarningr+r,r1r@postr- status_codedebuglenerrortextr= RequestErrorr/randomuniformtimesleepminr3) rrrRpayloadrUattemptdelayresponseexc sleep_times rrzBackendSpanExporter.exportXs[  F|  NM N N N FAA%AAA4.6t|55.&     ?-1->G) * < 5(, G$ %! 3 qLG N<,,V],^^'#--L!?SZZ!?!?!?@@@F(.444444444LcH22EC! 3s%2AD9DDE.E  Ec8|jdS)z!Close the underlying HTTP client.N)r@closerIs rrkzBackendSpanExporter.closes r)NNNr%r&r'r()r)r*r+r*r,r*r-r.r/r0r1r2r3r2)r)r.r) rrr rArCrr)r+r,rrkr"rrr$r$s##'"A T T T T TD    AA_AEE_EDD_D;3;3;3;3zrr$cdeZdZdZ d#d$dZd%dZd%dZd&dZd&dZd'd(dZ dZ dZ d)d*d"Z dS)+BatchTraceProcessorzSome implementation notes: 1. Using Queue, which is thread-safe. 2. Using a background thread to export spans, to minimize any performance issues. 3. Spans are stored in memory until they are exported. r6ffffff?exporterr max_queue_sizer0max_batch_sizeschedule_delayr2export_trigger_ratioc||_tj||_||_||_||_tj|_ t||z|_ tj |jz|_ tj|jd|_|jdS)a Args: exporter: The exporter to use. max_queue_size: The maximum number of spans to store in the queue. After this, we will start dropping spans. max_batch_size: The maximum number of spans to export in a single batch. schedule_delay: The delay between checks for new spans to export. export_trigger_ratio: The ratio of the queue size at which we will trigger an export. )maxsizeT)targetdaemonN) _exporterqueueQueue_queue_max_queue_size_max_batch_size_schedule_delay threadingEvent_shutdown_eventr0_export_trigger_sizera_next_export_timeThread_run_worker_threadstart)rrqrrrsrtrus rrAzBatchTraceProcessor.__init__s""6;k.6Y6Y6Y ---(00%(9M(M$N$N!"&t/C!C'.diMMM !!#####rtracer rrc |j|dS#tj$rt jdYdSwxYw)NzQueue is full, dropping trace.r} put_nowaitr{FullrrWrrs ron_trace_startz"BatchTraceProcessor.on_trace_startsZ = K " "5 ) ) ) ) )z = = = N; < < < < < < =#AAcdSNr"rs r on_trace_endz BatchTraceProcessor.on_trace_end rspan Span[Any]cdSrr"rrs r on_span_startz!BatchTraceProcessor.on_span_startrrc |j|dS#tj$rt jdYdSwxYw)NzQueue is full, dropping span.rrs r on_span_endzBatchTraceProcessor.on_span_endsZ < K " "4 ( ( ( ( (z < < < N: ; ; ; ; ; ; rs""""""  %%%%%% BBBBBBBBBBBBB/BBByyyyy/yyyxq3q3q3q3q3*q3q3q3j'&((''(899 r