o h,@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z ddl mZmZdd lmZdd lmZGd d d eZGd ddeZGdddeZeZeeZdddZdddZdS)) annotationsN)cached_property)Any)logger)TracingExporterTracingProcessor)Span)Tracec@seZdZdZd ddZdS) ConsoleSpanExporterz+Prints the traces and spans to the console.itemslist[Trace | Span[Any]]returnNonecCsD|D]}t|trtd|jd|jqtd|qdS)Nzxporter] Export trace_id=z, name=z[Exporter] Export span: ) isinstancer printtrace_idnameexport)selfr itemrZ/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/tracing/processors.pyrs  zConsoleSpanExporter.exportNr rrr)__name__ __module__ __qualname____doc__rrrrrr sr c@sfeZdZ       d#d$ddZd%ddZeddZeddZeddZd&dd Z d!d"Z dS)'BackendSpanExporterN'https://api.openai.com/v1/traces/ingest?>@api_key str | None organizationprojectendpointstr max_retriesint base_delayfloat max_delaycCsF||_||_||_||_||_||_||_tjtj dddd|_ 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. <@)timeoutconnectr1N) _api_key _organization_projectr(r*r,r.httpxClientTimeout_client)rr$r&r'r(r*r,r.rrr__init__szBackendSpanExporter.__init__cCsd|jvr |jd=||_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. r$N)__dict__r4)rr$rrr set_api_keyAs  zBackendSpanExporter.set_api_keycC|jptjdS)NOPENAI_API_KEY)r4osenvirongetrrrrr$OzBackendSpanExporter.api_keycCr>)N OPENAI_ORG_ID)r5r@rArBrCrrrr&SrDz BackendSpanExporter.organizationcCr>)NOPENAI_PROJECT_ID)r6r@rArBrCrrrr'WrDzBackendSpanExporter.projectr rrrc Cs|sdS|jstddSdd|D}d|i}d|jddd}|jr+|j|d <|jr3|j|d <d }|j} |d 7}zF|jj|j||d}|j dkr[t dt |dWdSd|j krfdkrxnnt d|j d|j WdStd|j dWntjy}ztd|WYd}~nd}~ww||jkrt ddS|td d|} t| t|d|j}q9)Nz0OPENAI_API_KEY is not set, skipping trace exportcSsg|] }|r|qSr)r).0rrrr csz.BackendSpanExporter.export..datazBearer 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'r,r:postr( status_codedebuglenerrortextr7 RequestErrorr*randomuniformtimesleepminr.) rr rIpayloadrLattemptdelayresponseexc sleep_timerrrr[sV         zBackendSpanExporter.exportcCs|jdS)z!Close the underlying HTTP client.N)r:closerCrrrraszBackendSpanExporter.close)NNNr r!r"r#)r$r%r&r%r'r%r(r)r*r+r,r-r.r-)r$r)r) rrrr;r=rr$r&r'rrarrrrrs$ "     =rc@s~eZdZdZ    d.d/ddZd0ddZd1ddZd1ddZd2ddZd2ddZ d3d4d#d$Z d%d&Z d'd(Z d5d6d,d-Z d S)7BatchTraceProcessorzSome 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. r0ffffff?exporterrmax_queue_sizer+max_batch_sizeschedule_delayr-export_trigger_ratiocCsh||_tj|d|_||_||_||_t|_ t dt |||_ t |j|_d|_t|_dS)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. )maxsizerN) _exporterqueueQueue_queue_max_queue_size_max_batch_size_schedule_delay threadingEvent_shutdown_eventmaxr+_export_trigger_sizerX_next_export_time_worker_threadLock_thread_start_lock)rrfrgrhrirjrrrr;s zBatchTraceProcessor.__init__rrcCs~|jr |jr dS|j(|jr|jr WddStj|jdd|_|jWddS1s8wYdS)NT)targetdaemon)ryis_aliver{rsThread_runstartrCrrr_ensure_thread_starteds "z*BatchTraceProcessor._ensure_thread_startedtracer cC<|z |j|WdStjytdYdSw)NzQueue is full, dropping trace.rro put_nowaitrmFullrrNrrrrron_trace_start z"BatchTraceProcessor.on_trace_startcCdSNrrrrr on_trace_endz BatchTraceProcessor.on_trace_endspan Span[Any]cCrrrrrrrr on_span_startrz!BatchTraceProcessor.on_span_startcCr)NzQueue is full, dropping span.rrrrr on_span_endrzBatchTraceProcessor.on_span_endNr1 float | NonecCs<|j|jr|jr|jj|ddS|jdddS)z` Called when the application stops. We signal our thread to stop, then join it. r3TforceN)rusetryr~join_export_batches)rr1rrrshutdowns zBatchTraceProcessor.shutdowncCs|jdddS)z@ Forces an immediate flush of all queued spans. TrN)rrCrrr force_flushszBatchTraceProcessor.force_flushcCsr|js1t}|j}||jks||jkr'|jddt|j|_nt d|jr|jdddS)NFrg?T) ruis_setrXroqsizerxrwrrrrY)r current_time queue_sizerrrrs     zBatchTraceProcessor._runFrboolcCs~ g}|js4|st||jkr4z ||jWn tjy%Ynw|js4|st||jks|s8dS|j |q)zDrains the queue and exports in batches. If force=True, export everything. Otherwise, export up to `max_batch_size` repeatedly until the queue is completely empty. TN) roemptyrRrqappend get_nowaitrmEmptyrlr)rritems_to_exportrrrrs&    z#BatchTraceProcessor._export_batches)rcrdr0re) rfrrgr+rhr+rir-rjr-)rr)rr rr)rrrrr)r1r)F)rr)rrrrr;rrrrrrrrrrrrrrbs   "    rbrcCtS)zOThe default exporter, which exports traces and spans to the backend in batches.)_global_exporterrrrrdefault_exporter/rrcCr)zPThe default processor, which exports traces and spans to the backend in batches.)_global_processorrrrrdefault_processor4rr)rr)rrb) __future__rr@rmrVrsrX functoolsrtypingrr7rprocessor_interfacerr spansr tracesr r rrbrrrrrrrrs*