؎j0UdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z m Z ddlmZddlmZdae gefed <d ae gefed <d e gefd dfdZd e gefd dfdZd efdZd efdZded efdZ d+dededzd efdZd,ded efdZde d efdZde de de d e fdZ d-d ed!e!d efd"Z"d#ed efd$Z#ded efd%Z$d#ed e!fd&Z%d'e!d e!fd(Z&d)e'd efd*Z(dS).z,The common utilities for agentscope library.N)datetime)AnyCallable)logger)ToolJSONDecodeErrorc2tjjSN)uuiduuid4hexGC:\Projects\DSCons\.venv\Lib\site-packages\agentscope\_utils\_common.pyrs)9r _id_factorycLtjSr )rnow isoformatrrrrrs 0H0H0J0Jr_timestamp_factoryfactoryreturncpt|s$tdt|j|adS)aeOverride the global ID factory used by all AgentScope entities. Entity IDs default to ``uuid.uuid4().hex``. Call this once at startup to substitute a different strategy. .. note:: Security-sensitive tokens (gateway tokens, Redis lock tokens) are **not** affected and always use ``uuid.uuid4().hex``. Args: factory (`Callable[[], str]`): A no-arg callable returning a string ID. Raises: TypeError: If ``factory`` is not callable. Example: >>> from agentscope import set_id_factory >>> set_id_factory(lambda: uuid7().hex) factory must be a callable, got N)callable TypeErrortype__name__rrs rset_id_factoryr sE* G    GtG}}/E G G   KKKrcpt|s$tdt|j|adS)zOverride the global timestamp factory used by all AgentScope entities. Args: factory (`Callable[[], str]`): A no-arg callable returning a string ID. Raises: TypeError: If ``factory`` is not callable. rN)rrrrrrs rset_timestamp_factoryr"4sH G    GtG}}/E G G   !rctS)z:Generate an ID string using the current global ID factory.)rrrr _generate_idr$Fs ==rctS)zKGenerate a timestamp string using the current global timestamp factory.)rrrr_generate_timestampr&Ks   rpathcztjtj|S)z=Expand user-home shorthand and return an absolute local path.)osr'abspath expanduser)r's r_normalize_local_pathr,Qs& 7??27--d33 4 44rjson_strschemac 2 tj|}t|tr|Sdt |d}n/#tj$r}dt |d}Yd}~nd}~wwxYw ddlm}||d| }tj|}t|tr|Sn#t$rYnwxYwt|d kr|dd d z|d dz}d}n|}d}td|d|dt|d)aThe given json_str maybe incomplete, e.g. '{"key', so we need to repair and load it into a Python object. .. note:: This function is currently only used for parsing the streaming output of the argument field in `tool_use`, so the parsed result must be a dict. Args: json_str (`str`): The JSON string to parse, which may be incomplete or malformed. schema (`dict`, optional): An optional JSON schema to guide the repair process. Returns: `dict`: A dictionary parsed from the JSON string after repair attempts. Returns an empty dict if all repair attempts fail. z.Error: Your argument string is decoded into a z' object, but a dict object is expected!z}Error: When decoding your tool arguments from JSON format to a Python dictionary, a JSONDecodeError was raised with message: .Nr) repair_jsonT) stream_stabler.dz [TRUNCATE]izm(Because the JSON string is too long, a truncated label "[TRUNCATE]" is used here to indicate the truncation)zz? Your argument string is decoded by the following code snippetz/: ```python import json your_tool_arguments = zn json.loads(your_tool_arguments) ``` **You should recorrect the arguments in JSON format.**) jsonloads isinstancedictrJSONDecodeErrorstr json_repairr1 Exceptionlenrrepr) r-r.res error_messageer1repairederror_json_str ellipsis_hints r_json_loads_with_repairrFVs. j"" c4  J 6T#YY 6 6 6        "A " " "     ++++++;xtFKKKj"" c4  J        8}}s!$3$,6$%%H D  "  Lm L L>K L L N++ L L L   s.*AAA-A((A-1>B11 B>=B>Fadd_random_suffixctjddd}|r,|dtjdz }|S)z@Get the current timestamp in the format YYYY-MM-DD HH:MM:SS.sss.z%Y-%m-%d %H:%M:%S.%fN_)rrstrftimer)urandomr )rG timestamps r_get_timestamprOsZ ''(>??DI/.A**,,... rfunccNKtj|ptj|p{t|tjpat|tjotj|p3t|tj ot|j d{VS)zCheck if the given function is an async function, including coroutine functions, async generators, and coroutine objects. N) inspectiscoroutinefunctionisasyncgenfunctionr8types CoroutineType GeneratorTypeasyncio iscoroutine functoolspartial_is_async_funcrP)rPs rr\r\s #D)) ,  %d + + , dE/ 0 0 , dE/ 0 0 &   % %  , dI- . . , ++ + + + + + +rargskwargsc\Kt|d{Vr||i|d{VS||i|S)aExecute an async or sync function based on its type. Args: func (`Callable`): The function to be executed, which can be either async or sync. *args (`Any`): Positional arguments to be passed to the function. **kwargs (`Any`): Keyword arguments to be passed to the function. Returns: `Any`: The result of the function execution. N)r\)rPr]r^s r_execute_async_or_sync_funcr`sf(D ! !!!!!!!+T4*6********* 4   rrKurl max_retriesc ddl}t|D]} ||}||jdcS#t $r1tj|jdcYcSt$r-}tj d|t|Yd}~d}~wwxYwtd|d|d) zGet the bytes from a given URL. Args: url (`str`): The URL to fetch the bytes from. max_retries (`int`, defaults to `3`): The maximum number of retries. rNutf-8asciiz8Failed to fetch bytes from URL %s. Error %s. Retrying...z Failed to fetch bytes from URL `z` after z retries.)requestsrangegetraise_for_statuscontentdecodeUnicodeDecodeErrorbase64 b64encoder=rinfor; RuntimeError)rarbrfrJresponserBs r_get_bytes_from_web_urlrrs1OOO ;   ||C((H  % % ' ' '#**733 3 3 3! F F F#H$455<`` or ``#/definitions/`` are expanded; external ``$ref`` URLs are left unchanged. Args: schema (`dict`): The JSON schema that may contain ``$defs`` and ``$ref`` references. Returns: `dict`: A flattened JSON schema with all references resolved inline. $defs definitionsobjvisitedrct|trfd|DSt|ts|Sd|vr|d}t|tr|ds|dr|dd}|vrt jd|dd |d d S| vrK ||hz}|D]\}}|dkr||hz||<|S|Si}|D]\}}|d vr |||<|S) Nc(g|]}|Srr).0item _resolve_refr}s r z>_flatten_json_schema.._resolve_ref..;s%@@@DLLw//@@@rz$refz#/$defs/z#/definitions//z3Circular reference detected for '%s' in tool schemaobjectz (circular: ))r description)rzr{) r8listr9r; startswithsplitrwarningitems) r|r}ref_pathdef_nameresolvedkeyvalueresultrdefss ` rrz*_flatten_json_schema.._resolve_ref9s c4  A@@@@@C@@@ @#t$$ J S==6{H(C(( $##J// $&&'788 $$>>#..r2w&&N!  !)'@X'@'@'@t##+|X8*,  H'*iikk U&==,8L % '8* 4--HSM$OJ!#))++ 7 7JC...&,ug66F3KK r) r8rhr9copydeepcopyupdatepop frozensetr)r.has_defsrrs @@r_flatten_json_schemar sC6&**W--t44  =!! 99H  ]6 " "FD&**W%%t,,) FJJw''(((&**]++T22/ FJJ}--...  4=KK((#( (C(((((((T <  rcltt|ddz dzS)z.Estimate the number of tokens in a given text.rdg?)intr>encoderws r_estimate_tokensrfs0 s4;;w''((1,s2 3 33rtokensc&t|dzS)z/Estimate the number of bytes with given tokens.r)r)rs r_estimate_bytesrls vz??rerrorct|tr$dd|jDSt |pt |jS)u.Render an exception as something a person can act on. Async transports run inside task groups, so what surfaces is often an ``ExceptionGroup`` whose own message is ``"unhandled errors in a TaskGroup (1 sub-exception)"`` — true, and of no use to anyone. The real cause is a leaf, so leaves are what get reported. Args: error (`BaseException`): The exception to describe. Returns: `str`: The leaf causes, joined; the exception type when a leaf carries no message of its own. z; c34K|]}t|VdSr )_describe_exception)rsubs r z&_describe_exception..s+NNc,S11NNNNNNr)r8BaseExceptionGroupjoin exceptionsr;rr)rs rrrrsV"%+,,OyyNNU=MNNNNNN u:: -e--rr )F)rK))__doc__rXrmrrZrRr6r)rUr rtypingrr_loggingr exceptionrrr;__annotations__rr r"r$r&r,r9rFboolrOr\r`rrrrxrrr BaseExceptionrrrrrsS222   ++++++":!9 Xb#g 999(J(JHRW%JJJHRW-$:!8BG#4!!!!!$c  S    555555NNN 4KN NNNNbds x D     ! ! !! !!!!8          F 5C 5C 5 5 5 5V V $V V V V r43434444 CC .}.......r