NgI FddlZddlmZddlmZd defdZd dZd ad Z dS) N) __version__) connectionutf-8strictreturnct|tr|St|tr|||St|SN) isinstancestrbytesdecode)sencodingerrorss V/var/www/html/ai-engine/env/lib/python3.11/site-packages/langsmith/_internal/_patch.py _ensure_strrsJ!S!U*xx&))) q66Mct|dd|j|j|i}n|}dd|Dvrt j|d<|||||S)aMake the request. This function is based on the urllib3 request method, with modifications to handle potential issues when using vcrpy in concurrent workloads. Args: self: The HTTPConnection instance. method (str): The HTTP method (e.g., 'GET', 'POST'). url (str): The URL for the request. body (Optional[Any]): The body of the request. headers (Optional[dict]): Headers to send with the request. Returns: The result of calling the parent request method. sockNz user-agentc3XK|]%}t|V&dSr )rlower).0ks r zrequest..+s2DDqK 22DDDDDDrz User-Agent)bodyheaders)getattrr settimeouttimeoutcopyr_get_default_user_agent_parent_request)selfmethodurlrrs rrequestr&s$tVT"". T\***,,..DDGDDDDD * B D D    $  H HHrFctrdSddlm}|t|dkrdadSt jjd}|jfd}|t j_dadS)apPatch the request method of urllib3 to avoid type errors when using vcrpy. In concurrent workloads (such as the tracing background queue), the connection pool can get in a state where an HTTPConnection is created before vcrpy patches the HTTPConnection class. In urllib3 >= 2.0 this isn't a problem since they use the proper super().request(...) syntax, but in older versions, super(HTTPConnection, self).request is used, resulting in a TypeError since self is no longer a subclass of "HTTPConnection" (which at this point is vcr.stubs.VCRConnection). This method patches the class to fix the super() syntax to avoid mixed inheritance. In the case of the LangSmith tracing logic, it doesn't really matter since we always exclude cache checks for calls to LangSmith. The patch is only applied for urllib3 versions older than 2.0. Nr)versionz2.0TcXtj||_t|g|Ri|S)zHandle parent request. This method binds the parent's request method to self and then calls our modified request function. ) functoolspartialr"r&)r#argskwargsparent_requests r new_requestz"patch_urllib3..new_requestSs8 )0FFt-d---f---r) _PATCHED packagingr(parseurllib3versionrHTTPConnection __bases__r&)r( parent_classr/r.s @r patch_urllib3r75s$!!!!!!}}^$$ e(<(<<<,6q9L!)N.....)4J%HHHr)rr)NN) r*urllib3rr3rr rr&r0r7rrr:s111111IIIIB (((((r