Remove unused msgpack import

[creatorrr]

trafficstars

Summary

• delete unused imports from agents_api/init.py

Testing

• python -m ward --help (fails: No module named ward)
• poe test (fails: command not found)

[qodo-merge-for-open-source[bot]]

CI Feedback 🧐

(Feedback updated until commit https://github.com/julep-ai/julep/commit/c84c2c1b65cd7624aa4ee673fbc2a410b023375b)

A test triggered by this PR failed. Here is an AI-generated analysis of the failure:

Action: Test

Failed stage: [❌]

Failed test name: test_docs_queries

Failure summary: The action failed because three tests in test_docs_queries.py failed: 1. test_docs_queries:619 query: search docs by embedding 2. test_docs_queries:647 query: search docs by hybrid 3. query: get doc (line number not specified) All failures show the same root cause: database connection issues with asyncpg. The error messages indicate: "RuntimeError: Event loop is closed" "RuntimeError: unable to perform operation on <TCPTransport closed=True reading=False>; the handler is closed" "InterfaceError: cannot perform operation: another operation is in progress" This suggests a concurrency issue where database connections are being closed while operations are still in progress, or multiple operations are being attempted on the same connection simultaneously.

Relevant error logs: 1: ##[group]Operating System 2: Ubuntu ... 956: prune-cache: true 957: ignore-nothing-to-cache: false 958: ##[endgroup] 959: Downloading uv from "https://github.com/astral-sh/uv/releases/download/0.7.5/uv-x86_64-unknown-linux-gnu.tar.gz" ... 960: [command]/usr/bin/tar xz --warning=no-unknown-keyword --overwrite -C /home/runner/work/_temp/4a00088e-f741-4f99-81d6-1c4f0294fdd1 -f /home/runner/work/_temp/cd8b4d61-f7cc-4755-960e-c87a50024dcc 961: Added /opt/hostedtoolcache/uv/0.7.5/x86_64 to the path 962: Added /home/runner/.local/bin to the path 963: Set UV_CACHE_DIR to /home/runner/work/_temp/setup-uv-cache 964: Successfully installed uv version 0.7.5 965: Searching files using cache dependency glob: **/uv.lock 966: /home/runner/work/julep/julep/agents-api/uv.lock 967: /home/runner/work/julep/julep/cli/uv.lock 968: /home/runner/work/julep/julep/integrations-service/uv.lock 969: Found 3 files to hash. 970: Trying to restore uv cache from GitHub Actions cache with key: setup-uv-1-x86_64-unknown-linux-gnu-0.7.5-d92603d25acef1c08e643c37cc2475e5e190deb9690356b084828d60043a591f 971: ##[warning]Failed to restore: Cache service responded with 422 972: No GitHub Actions cache found for key: setup-uv-1-x86_64-unknown-linux-gnu-0.7.5-d92603d25acef1c08e643c37cc2475e5e190deb9690356b084828d60043a591f ... 1479: sql 1480: PASS test_agent_queries:126 query: update agent sql 4% 1481: PASS test_agent_queries:153 query: update agent with project sql 4% 1482: PASS test_agent_queries:177 query: update agent, project does not exist 4% 1483: PASS test_agent_queries:201 query: patch agent sql 5% 1484: PASS test_agent_queries:225 query: patch agent with project sql 5% 1485: PASS test_agent_queries:260 query: patch agent, project does not exist 5% 1486: PASS test_agent_queries:283 query: get agent not exists sql 5% 1487: PASS test_agent_queries:294 query: get agent exists sql 6% 1488: PASS test_agent_queries:315 query: list agents sql 6% 1489: PASS test_agent_queries:326 query: list agents with project filter sql 6% 1490: PASS test_agent_queries:352 query: list agents sql, invalid sort 6% 1491: direction 1492: PASS test_agent_queries:368 query: delete agent sql 6% 1493: INFO:httpx:HTTP Request: POST http://testserver/agents "HTTP/1.1 403 Forbidden" 1494: PASS test_agent_routes:9 route: unauthorized should fail 7% 1495: INFO:httpx:HTTP Request: POST http://testserver/agents "HTTP/1.1 201 Created" ... 1580: PASS test_docs_queries:255 query: list user docs, invalid sort by 18% 1581: PASS test_docs_queries:285 query: list user docs, invalid sort direction 19% 1582: PASS test_docs_queries:315 query: list agent docs 19% 1583: PASS test_docs_queries:371 query: list agent docs, invalid limit 19% 1584: PASS test_docs_queries:390 query: list agent docs, invalid offset 19% 1585: PASS test_docs_queries:409 query: list agent docs, invalid sort by 20% 1586: PASS test_docs_queries:427 query: list agent docs, invalid sort direction 20% 1587: PASS test_docs_queries:445 query: delete user doc 20% 1588: PASS test_docs_queries:482 query: delete agent doc 20% 1589: PASS test_docs_queries:519 query: search docs by text 20% 1590: PASS test_docs_queries:556 query: search docs by text with technical 21% 1591: terms and phrases 1592: FAIL test_docs_queries:619 query: search docs by embedding 21% 1593: FAIL test_docs_queries:647 query: search docs by hybrid 21% 1594: ──────────────────────────────── query: get doc ──────────────────────────────── 1595: Failed at tests/test_docs_queries.py 1596: ╭─────────────────── Traceback (most recent call last) ────────────────────╮ 1597: │ in uvloop.loop.Loop.call_soon:1281 │ 1598: │ │ 1599: │ in uvloop.loop.Loop._call_soon:669 │ 1600: │ │ 1601: │ in uvloop.loop.UVStream.write:678 │ 1602: │ │ 1603: │ in uvloop.loop.Loop._append_ready_handle:673 │ 1604: │ │ 1605: │ in uvloop.loop.Loop._check_closed:705 │ 1606: ╰──────────────────────────────────────────────────────────────────────────╯ 1607: RuntimeError: Event loop is closed 1608: During handling of the above exception, another exception occurred: ... 1649: │ │ │ │ │ │ │ │ 1650: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 1651: │ │ │ │ │ │ │ │ 1652: │ │ UUID('06828a99-03b2-7841-8000-0fa99df47a0c'), │ │ 1653: │ │ │ │ │ │ 'agent', │ │ 1654: │ │ │ │ │ │ │ │ 1655: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4') │ │ 1656: │ │ │ │ │ ], │ │ 1657: │ │ │ │ │ 'timeout': 90.0 │ │ 1658: │ │ │ │ } │ │ 1659: │ │ │ ) │ │ 1660: │ │ ] │ │ 1661: │ │ conn = <PoolConnectionProxy │ │ 1662: │ │ <asyncpg.connection.Connection object at │ │ 1663: │ │ 0x7ff3c23e37a0> 0x7ff3a331ad70> │ │ 1664: │ │ connection_error = False │ │ 1665: │ │ connection_pool = <asyncpg.pool.Pool object at 0x7ff38c967f40> │ │ 1666: │ │ debug = None │ │ 1667: │ │ kwargs = { │ │ 1668: │ │ │ 'developer_id': │ │ 1669: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 1670: │ │ │ 'doc_id': │ │ 1671: │ │ UUID('06828a99-03b2-7841-8000-0fa99df47a0c'), │ │ 1672: │ │ │ 'owner_type': 'agent', │ │ 1673: │ │ │ 'owner_id': │ │ 1674: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4') │ │ 1675: │ │ } │ │ 1676: │ │ only_on_error = False │ │ 1677: │ │ pool = <asyncpg.pool.Pool object at 0x7ff38c967f40> │ │ ... 1700: │ │ │ │ │ │ │ 1701: │ │ UUID('06828a99-03b2-7841-8000-0fa99df47a0c'), │ │ 1702: │ │ │ │ │ 'agent', │ │ 1703: │ │ │ │ │ │ │ 1704: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4') │ │ 1705: │ │ │ │ ] │ │ 1706: │ │ │ ) │ │ 1707: │ │ ] │ │ 1708: │ │ return_index = -1 │ │ 1709: │ │ timeit = False │ │ 1710: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 1711: │ │ 1712: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 1713: │ ges/asyncpg/transaction.py:68 in __aenter__ │ 1714: │ │ 1715: │ 65 │ │ │ raise apg_errors.InterfaceError( │ 1716: │ 66 │ │ │ │ 'cannot enter context: already in an `async with` │ 1717: │ 67 │ │ self._managed = True │ 1718: │ ❱ 68 │ │ await self.start() │ 1719: │ 69 │ │ 1720: │ 70 │ async def __aexit__(self, extype, ex, tb): │ 1721: │ 71 │ │ try: │ 1722: │ │ 1723: │ ╭───────────────────────── locals ─────────────────────────╮ │ 1724: │ │ self = <asyncpg.Transaction state:failed 0x7ff3a048cf90> │ │ 1725: │ ╰──────────────────────────────────────────────────────────╯ │ 1726: │ │ 1727: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 1728: │ ges/asyncpg/transaction.py:146 in start │ 1729: │ │ 1730: │ 143 │ │ │ query += ';' │ 1731: │ 144 │ │ │ 1732: │ 145 │ │ try: │ 1733: │ ❱ 146 │ │ │ await self._connection.execute(query) │ 1734: │ 147 │ │ except BaseException: │ 1735: │ 148 │ │ │ self._state = TransactionState.FAILED │ 1736: │ 149 │ │ │ raise │ 1737: │ │ 1738: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 1739: │ │ con = <asyncpg.connection.Connection object at 0x7ff3c23e37a0> │ │ 1740: │ │ query = 'BEGIN;' │ │ 1741: │ │ self = <asyncpg.Transaction state:failed 0x7ff3a048cf90> │ │ 1742: │ ╰──────────────────────────────────────────────────────────────────╯ │ ... 1759: │ │ timeout = None │ │ 1760: │ ╰────────────────────────────────────────────────────────────────────╯ │ 1761: │ │ 1762: │ in query:375 │ 1763: │ │ 1764: │ in asyncpg.protocol.protocol.BaseProtocol.query:368 │ 1765: │ │ 1766: │ in asyncpg.protocol.protocol.CoreProtocol._simple_query:1174 │ 1767: │ │ 1768: │ in asyncpg.protocol.protocol.BaseProtocol._write:967 │ 1769: │ │ 1770: │ in uvloop.loop.UVStream.write:678 │ 1771: │ │ 1772: │ in uvloop.loop.UVHandle._ensure_alive:159 │ 1773: ╰──────────────────────────────────────────────────────────────────────────╯ 1774: RuntimeError: unable to perform operation on <TCPTransport closed=True 1775: reading=False 0x416336b0>; the handler is closed ... 1788: │ │ 1789: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 1790: │ │ debug = None │ │ 1791: │ │ loop_factory = None │ │ 1792: │ │ main = <async_generator_asend object at 0x7ff3c4bf19c0> │ │ 1793: │ │ runner = <asyncio.runners.Runner object at 0x7ff3a331b050> │ │ 1794: │ ╰──────────────────────────────────────────────────────────────────╯ │ 1795: │ │ 1796: │ /home/runner/.local/share/uv/python/cpython-3.12.10-linux-x86_64-gnu/lib │ 1797: │ /python3.12/asyncio/runners.py:118 in run │ 1798: │ │ 1799: │ 115 │ │ │ 1800: │ 116 │ │ self._interrupt_count = 0 │ 1801: │ 117 │ │ try: │ 1802: │ ❱ 118 │ │ │ return self._loop.run_until_complete(task) │ 1803: │ 119 │ │ except exceptions.CancelledError: │ 1804: │ 120 │ │ │ if self._interrupt_count > 0: │ 1805: │ 121 │ │ │ │ uncancel = getattr(task, "uncancel", None) │ 1806: │ │ 1807: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 1808: │ │ context = <_contextvars.Context object at 0x7ff38f274400> │ │ 1809: │ │ coro = <async_generator_asend object at 0x7ff3c4bf19c0> │ │ 1810: │ │ self = <asyncio.runners.Runner object at 0x7ff3a331b050> │ │ 1811: │ │ sigint_handler = functools.partial(<bound method Runner._on_sigint │ │ 1812: │ │ of <asyncio.runners.Runner object at │ │ 1813: │ │ 0x7ff3a331b050>>, main_task=<Task finished │ │ 1814: │ │ name='Task-1627' coro=<<async_generator_asend │ │ 1815: │ │ without __name__>()> │ │ 1816: │ │ exception=InterfaceError('cannot perform operation: │ │ 1817: │ │ another operation is in progress')>) │ │ 1818: │ │ task = <Task finished name='Task-1627' │ │ 1819: │ │ coro=<<async_generator_asend without __name__>()> │ │ 1820: │ │ exception=InterfaceError('cannot perform operation: │ │ 1821: │ │ another operation is in progress')> │ │ ... 1871: │ │ 52, 231222, tzinfo=datetime.timezone.utc), │ │ 1872: │ │ │ title='Hello', │ │ 1873: │ │ │ content=['World', 'World2', 'World3'], │ │ 1874: │ │ │ embeddings=[], │ │ 1875: │ │ │ modality='text', │ │ 1876: │ │ │ language='english', │ │ 1877: │ │ │ embedding_model='text-embedding-3-large', │ │ 1878: │ │ │ embedding_dimensions=1024 │ │ 1879: │ │ ) │ │ 1880: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 1881: │ │ 1882: │ /home/runner/work/julep/julep/agents-api/agents_api/queries/utils.py:320 │ 1883: │ in async_wrapper │ 1884: │ │ 1885: │ 317 │ │ │ │ result: T = await func(*args, **kwargs) │ 1886: │ 318 │ │ │ except BaseException as error: │ 1887: │ 319 │ │ │ │ _check_error(error) │ 1888: │ ❱ 320 │ │ │ │ raise error │ 1889: │ 321 │ │ │ │ ... 1899: │ │ │ 'owner_type': 'agent', │ │ 1900: │ │ │ 'owner_id': │ │ 1901: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4'), │ │ 1902: │ │ │ 'connection_pool': <asyncpg.pool.Pool object at │ │ 1903: │ │ 0x7ff38c967f40> │ │ 1904: │ │ } │ │ 1905: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 1906: │ │ 1907: │ /home/runner/work/julep/julep/agents-api/agents_api/queries/utils.py:317 │ 1908: │ in async_wrapper │ 1909: │ │ 1910: │ 314 │ │ @wraps(func) │ 1911: │ 315 │ │ async def async_wrapper(*args: P.args, **kwargs: P.kwargs) │ 1912: │ 316 │ │ │ try: │ 1913: │ ❱ 317 │ │ │ │ result: T = await func(*args, **kwargs) │ 1914: │ 318 │ │ │ except BaseException as error: │ 1915: │ 319 │ │ │ │ _check_error(error) │ 1916: │ 320 │ │ │ │ raise error │ 1917: │ │ ... 1996: │ │ │ │ │ │ │ │ 1997: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 1998: │ │ │ │ │ │ │ │ 1999: │ │ UUID('06828a99-03b2-7841-8000-0fa99df47a0c'), │ │ 2000: │ │ │ │ │ │ 'agent', │ │ 2001: │ │ │ │ │ │ │ │ 2002: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4') │ │ 2003: │ │ │ │ │ ], │ │ 2004: │ │ │ │ │ 'timeout': 90.0 │ │ 2005: │ │ │ │ } │ │ 2006: │ │ │ ) │ │ 2007: │ │ ] │ │ 2008: │ │ conn = <PoolConnectionProxy │ │ 2009: │ │ <asyncpg.connection.Connection object at │ │ 2010: │ │ 0x7ff3c23e37a0> 0x7ff3a331ad70> │ │ 2011: │ │ connection_error = False │ │ 2012: │ │ connection_pool = <asyncpg.pool.Pool object at 0x7ff38c967f40> │ │ 2013: │ │ debug = None │ │ 2014: │ │ kwargs = { │ │ 2015: │ │ │ 'developer_id': │ │ 2016: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2017: │ │ │ 'doc_id': │ │ 2018: │ │ UUID('06828a99-03b2-7841-8000-0fa99df47a0c'), │ │ 2019: │ │ │ 'owner_type': 'agent', │ │ 2020: │ │ │ 'owner_id': │ │ 2021: │ │ UUID('06828a99-003d-7f2d-8000-0c83d70a2ac4') │ │ 2022: │ │ } │ │ 2023: │ │ only_on_error = False │ │ 2024: │ │ pool = <asyncpg.pool.Pool object at 0x7ff38c967f40> │ │ ... 2059: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2060: │ ges/asyncpg/pool.py:1031 in __aexit__ │ 2061: │ │ 2062: │ 1028 │ │ self.done = True │ 2063: │ 1029 │ │ con = self.connection │ 2064: │ 1030 │ │ self.connection = None │ 2065: │ ❱ 1031 │ │ await self.pool.release(con) │ 2066: │ 1032 │ │ 2067: │ 1033 │ def __await__(self): │ 2068: │ 1034 │ │ self.done = True │ 2069: │ │ 2070: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2071: │ │ con = <PoolConnectionProxy <asyncpg.connection.Connection object at │ │ 2072: │ │ 0x7ff3c23e37a0> 0x7ff3a331ad70> │ │ 2073: │ │ exc = ( │ │ 2074: │ │ │ <class 'RuntimeError'>, │ │ 2075: │ │ │ RuntimeError('unable to perform operation on │ │ 2076: │ │ <TCPTransport closed=True reading=False 0x416336b0>; the │ │ ... 2091: │ 907 │ async def close(self): │ 2092: │ 908 │ │ """Attempt to gracefully close all connections in the poo │ 2093: │ │ 2094: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2095: │ │ ch = <asyncpg.pool.PoolConnectionHolder object at │ │ 2096: │ │ 0x7ff38edf14c0> │ │ 2097: │ │ connection = <PoolConnectionProxy <asyncpg.connection.Connection │ │ 2098: │ │ object at 0x7ff3c23e37a0> 0x7ff3a331ad70> │ │ 2099: │ │ self = <asyncpg.pool.Pool object at 0x7ff38c967f40> │ │ 2100: │ │ timeout = None │ │ 2101: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2102: │ │ 2103: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2104: │ ges/asyncpg/pool.py:228 in release │ 2105: │ │ 2106: │ 225 │ │ │ │ # an IO error, so terminate the connection. │ 2107: │ 226 │ │ │ │ self._con.terminate() │ ... 2113: │ │ 2114: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2115: │ │ budget = None │ │ 2116: │ │ self = <asyncpg.pool.PoolConnectionHolder object at │ │ 2117: │ │ 0x7ff38edf14c0> │ │ 2118: │ │ timeout = None │ │ 2119: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2120: │ │ 2121: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2122: │ ges/asyncpg/pool.py:218 in release │ 2123: │ │ 2124: │ 215 │ │ │ │ │ await self._con._reset() │ 2125: │ 216 │ │ │ │ │ await self._pool._reset(self._con) │ 2126: │ 217 │ │ │ else: │ 2127: │ ❱ 218 │ │ │ │ await self._con.reset(timeout=budget) │ 2128: │ 219 │ │ except (Exception, asyncio.CancelledError) as ex: │ 2129: │ 220 │ │ │ # If the `reset` call failed, terminate the connectio │ 2130: │ 221 │ │ │ # A new one will be created when `acquire` is called │ ... 2177: │ 350 │ │ │ return result │ 2178: │ 351 │ │ │ 2179: │ 352 │ │ _, status, _ = await self._execute( │ 2180: │ │ 2181: │ ╭────────────────────────────── locals ──────────────────────────────╮ │ 2182: │ │ args = () │ │ 2183: │ │ query = 'ROLLBACK' │ │ 2184: │ │ self = <asyncpg.connection.Connection object at 0x7ff3c23e37a0> │ │ 2185: │ │ timeout = None │ │ 2186: │ ╰────────────────────────────────────────────────────────────────────╯ │ 2187: │ │ 2188: │ in query:360 │ 2189: │ │ 2190: │ in asyncpg.protocol.protocol.BaseProtocol._check_state:745 │ 2191: ╰──────────────────────────────────────────────────────────────────────────╯ 2192: InterfaceError: cannot perform operation: another operation is in progress 2193: ─────────────────────── query: search docs by embedding ──────────────────────── 2194: Failed at tests/test_docs_queries.py 2195: ╭─────────────────── Traceback (most recent call last) ────────────────────╮ 2196: │ in uvloop.loop.Loop.call_soon:1281 │ 2197: │ │ 2198: │ in uvloop.loop.Loop._call_soon:669 │ 2199: │ │ 2200: │ in uvloop.loop.UVStream.write:678 │ 2201: │ │ 2202: │ in uvloop.loop.Loop._append_ready_handle:673 │ 2203: │ │ 2204: │ in uvloop.loop.Loop._check_closed:705 │ 2205: ╰──────────────────────────────────────────────────────────────────────────╯ 2206: RuntimeError: Event loop is closed 2207: During handling of the above exception, another exception occurred: ... 2248: │ │ │ │ │ │ │ │ 2249: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2250: │ │ │ │ │ │ │ │ 2251: │ │ UUID('06828a99-7713-72ab-8000-ddcc66b687a9'), │ │ 2252: │ │ │ │ │ │ 'agent', │ │ 2253: │ │ │ │ │ │ │ │ 2254: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3') │ │ 2255: │ │ │ │ │ ], │ │ 2256: │ │ │ │ │ 'timeout': 90.0 │ │ 2257: │ │ │ │ } │ │ 2258: │ │ │ ) │ │ 2259: │ │ ] │ │ 2260: │ │ conn = <PoolConnectionProxy │ │ 2261: │ │ <asyncpg.connection.Connection object at │ │ 2262: │ │ 0x7ff3c23e1c70> 0x7ff3a3360ca0> │ │ 2263: │ │ connection_error = False │ │ 2264: │ │ connection_pool = <asyncpg.pool.Pool object at 0x7ff38c964c40> │ │ 2265: │ │ debug = None │ │ 2266: │ │ kwargs = { │ │ 2267: │ │ │ 'developer_id': │ │ 2268: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2269: │ │ │ 'doc_id': │ │ 2270: │ │ UUID('06828a99-7713-72ab-8000-ddcc66b687a9'), │ │ 2271: │ │ │ 'owner_type': 'agent', │ │ 2272: │ │ │ 'owner_id': │ │ 2273: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3') │ │ 2274: │ │ } │ │ 2275: │ │ only_on_error = False │ │ 2276: │ │ pool = <asyncpg.pool.Pool object at 0x7ff38c964c40> │ │ ... 2299: │ │ │ │ │ │ │ 2300: │ │ UUID('06828a99-7713-72ab-8000-ddcc66b687a9'), │ │ 2301: │ │ │ │ │ 'agent', │ │ 2302: │ │ │ │ │ │ │ 2303: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3') │ │ 2304: │ │ │ │ ] │ │ 2305: │ │ │ ) │ │ 2306: │ │ ] │ │ 2307: │ │ return_index = -1 │ │ 2308: │ │ timeit = False │ │ 2309: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2310: │ │ 2311: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2312: │ ges/asyncpg/transaction.py:68 in __aenter__ │ 2313: │ │ 2314: │ 65 │ │ │ raise apg_errors.InterfaceError( │ 2315: │ 66 │ │ │ │ 'cannot enter context: already in an `async with` │ 2316: │ 67 │ │ self._managed = True │ 2317: │ ❱ 68 │ │ await self.start() │ 2318: │ 69 │ │ 2319: │ 70 │ async def __aexit__(self, extype, ex, tb): │ 2320: │ 71 │ │ try: │ 2321: │ │ 2322: │ ╭───────────────────────── locals ─────────────────────────╮ │ 2323: │ │ self = <asyncpg.Transaction state:failed 0x7ff38fcc78b0> │ │ 2324: │ ╰──────────────────────────────────────────────────────────╯ │ 2325: │ │ 2326: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2327: │ ges/asyncpg/transaction.py:146 in start │ 2328: │ │ 2329: │ 143 │ │ │ query += ';' │ 2330: │ 144 │ │ │ 2331: │ 145 │ │ try: │ 2332: │ ❱ 146 │ │ │ await self._connection.execute(query) │ 2333: │ 147 │ │ except BaseException: │ 2334: │ 148 │ │ │ self._state = TransactionState.FAILED │ 2335: │ 149 │ │ │ raise │ 2336: │ │ 2337: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 2338: │ │ con = <asyncpg.connection.Connection object at 0x7ff3c23e1c70> │ │ 2339: │ │ query = 'BEGIN;' │ │ 2340: │ │ self = <asyncpg.Transaction state:failed 0x7ff38fcc78b0> │ │ 2341: │ ╰──────────────────────────────────────────────────────────────────╯ │ ... 2358: │ │ timeout = None │ │ 2359: │ ╰────────────────────────────────────────────────────────────────────╯ │ 2360: │ │ 2361: │ in query:375 │ 2362: │ │ 2363: │ in asyncpg.protocol.protocol.BaseProtocol.query:368 │ 2364: │ │ 2365: │ in asyncpg.protocol.protocol.CoreProtocol._simple_query:1174 │ 2366: │ │ 2367: │ in asyncpg.protocol.protocol.BaseProtocol._write:967 │ 2368: │ │ 2369: │ in uvloop.loop.UVStream.write:678 │ 2370: │ │ 2371: │ in uvloop.loop.UVHandle._ensure_alive:159 │ 2372: ╰──────────────────────────────────────────────────────────────────────────╯ 2373: RuntimeError: unable to perform operation on <TCPTransport closed=True 2374: reading=False 0x4147f800>; the handler is closed ... 2387: │ │ 2388: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 2389: │ │ debug = None │ │ 2390: │ │ loop_factory = None │ │ 2391: │ │ main = <async_generator_asend object at 0x7ff3c5388900> │ │ 2392: │ │ runner = <asyncio.runners.Runner object at 0x7ff3a3360470> │ │ 2393: │ ╰──────────────────────────────────────────────────────────────────╯ │ 2394: │ │ 2395: │ /home/runner/.local/share/uv/python/cpython-3.12.10-linux-x86_64-gnu/lib │ 2396: │ /python3.12/asyncio/runners.py:118 in run │ 2397: │ │ 2398: │ 115 │ │ │ 2399: │ 116 │ │ self._interrupt_count = 0 │ 2400: │ 117 │ │ try: │ 2401: │ ❱ 118 │ │ │ return self._loop.run_until_complete(task) │ 2402: │ 119 │ │ except exceptions.CancelledError: │ 2403: │ 120 │ │ │ if self._interrupt_count > 0: │ 2404: │ 121 │ │ │ │ uncancel = getattr(task, "uncancel", None) │ 2405: │ │ 2406: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2407: │ │ context = <_contextvars.Context object at 0x7ff38f275bc0> │ │ 2408: │ │ coro = <async_generator_asend object at 0x7ff3c5388900> │ │ 2409: │ │ self = <asyncio.runners.Runner object at 0x7ff3a3360470> │ │ 2410: │ │ sigint_handler = functools.partial(<bound method Runner._on_sigint │ │ 2411: │ │ of <asyncio.runners.Runner object at │ │ 2412: │ │ 0x7ff3a3360470>>, main_task=<Task finished │ │ 2413: │ │ name='Task-2089' coro=<<async_generator_asend │ │ 2414: │ │ without __name__>()> │ │ 2415: │ │ exception=InterfaceError('cannot perform operation: │ │ 2416: │ │ another operation is in progress')>) │ │ 2417: │ │ task = <Task finished name='Task-2089' │ │ 2418: │ │ coro=<<async_generator_asend without __name__>()> │ │ 2419: │ │ exception=InterfaceError('cannot perform operation: │ │ 2420: │ │ another operation is in progress')> │ │ ... 2470: │ │ 59, 442329, tzinfo=datetime.timezone.utc), │ │ 2471: │ │ │ title='Hello', │ │ 2472: │ │ │ content=['World', 'World2', 'World3'], │ │ 2473: │ │ │ embeddings=[], │ │ 2474: │ │ │ modality='text', │ │ 2475: │ │ │ language='english', │ │ 2476: │ │ │ embedding_model='text-embedding-3-large', │ │ 2477: │ │ │ embedding_dimensions=1024 │ │ 2478: │ │ ) │ │ 2479: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2480: │ │ 2481: │ /home/runner/work/julep/julep/agents-api/agents_api/queries/utils.py:320 │ 2482: │ in async_wrapper │ 2483: │ │ 2484: │ 317 │ │ │ │ result: T = await func(*args, **kwargs) │ 2485: │ 318 │ │ │ except BaseException as error: │ 2486: │ 319 │ │ │ │ _check_error(error) │ 2487: │ ❱ 320 │ │ │ │ raise error │ 2488: │ 321 │ │ │ │ ... 2498: │ │ │ 'owner_type': 'agent', │ │ 2499: │ │ │ 'owner_id': │ │ 2500: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3'), │ │ 2501: │ │ │ 'connection_pool': <asyncpg.pool.Pool object at │ │ 2502: │ │ 0x7ff38c964c40> │ │ 2503: │ │ } │ │ 2504: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2505: │ │ 2506: │ /home/runner/work/julep/julep/agents-api/agents_api/queries/utils.py:317 │ 2507: │ in async_wrapper │ 2508: │ │ 2509: │ 314 │ │ @wraps(func) │ 2510: │ 315 │ │ async def async_wrapper(*args: P.args, **kwargs: P.kwargs) │ 2511: │ 316 │ │ │ try: │ 2512: │ ❱ 317 │ │ │ │ result: T = await func(*args, **kwargs) │ 2513: │ 318 │ │ │ except BaseException as error: │ 2514: │ 319 │ │ │ │ _check_error(error) │ 2515: │ 320 │ │ │ │ raise error │ 2516: │ │ ... 2595: │ │ │ │ │ │ │ │ 2596: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2597: │ │ │ │ │ │ │ │ 2598: │ │ UUID('06828a99-7713-72ab-8000-ddcc66b687a9'), │ │ 2599: │ │ │ │ │ │ 'agent', │ │ 2600: │ │ │ │ │ │ │ │ 2601: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3') │ │ 2602: │ │ │ │ │ ], │ │ 2603: │ │ │ │ │ 'timeout': 90.0 │ │ 2604: │ │ │ │ } │ │ 2605: │ │ │ ) │ │ 2606: │ │ ] │ │ 2607: │ │ conn = <PoolConnectionProxy │ │ 2608: │ │ <asyncpg.connection.Connection object at │ │ 2609: │ │ 0x7ff3c23e1c70> 0x7ff3a3360ca0> │ │ 2610: │ │ connection_error = False │ │ 2611: │ │ connection_pool = <asyncpg.pool.Pool object at 0x7ff38c964c40> │ │ 2612: │ │ debug = None │ │ 2613: │ │ kwargs = { │ │ 2614: │ │ │ 'developer_id': │ │ 2615: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2616: │ │ │ 'doc_id': │ │ 2617: │ │ UUID('06828a99-7713-72ab-8000-ddcc66b687a9'), │ │ 2618: │ │ │ 'owner_type': 'agent', │ │ 2619: │ │ │ 'owner_id': │ │ 2620: │ │ UUID('06828a99-73aa-787a-8000-7251da80ccc3') │ │ 2621: │ │ } │ │ 2622: │ │ only_on_error = False │ │ 2623: │ │ pool = <asyncpg.pool.Pool object at 0x7ff38c964c40> │ │ ... 2658: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2659: │ ges/asyncpg/pool.py:1031 in __aexit__ │ 2660: │ │ 2661: │ 1028 │ │ self.done = True │ 2662: │ 1029 │ │ con = self.connection │ 2663: │ 1030 │ │ self.connection = None │ 2664: │ ❱ 1031 │ │ await self.pool.release(con) │ 2665: │ 1032 │ │ 2666: │ 1033 │ def __await__(self): │ 2667: │ 1034 │ │ self.done = True │ 2668: │ │ 2669: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2670: │ │ con = <PoolConnectionProxy <asyncpg.connection.Connection object at │ │ 2671: │ │ 0x7ff3c23e1c70> 0x7ff3a3360ca0> │ │ 2672: │ │ exc = ( │ │ 2673: │ │ │ <class 'RuntimeError'>, │ │ 2674: │ │ │ RuntimeError('unable to perform operation on │ │ 2675: │ │ <TCPTransport closed=True reading=False 0x4147f800>; the │ │ ... 2690: │ 907 │ async def close(self): │ 2691: │ 908 │ │ """Attempt to gracefully close all connections in the poo │ 2692: │ │ 2693: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2694: │ │ ch = <asyncpg.pool.PoolConnectionHolder object at │ │ 2695: │ │ 0x7ff38edb2c40> │ │ 2696: │ │ connection = <PoolConnectionProxy <asyncpg.connection.Connection │ │ 2697: │ │ object at 0x7ff3c23e1c70> 0x7ff3a3360ca0> │ │ 2698: │ │ self = <asyncpg.pool.Pool object at 0x7ff38c964c40> │ │ 2699: │ │ timeout = None │ │ 2700: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2701: │ │ 2702: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2703: │ ges/asyncpg/pool.py:228 in release │ 2704: │ │ 2705: │ 225 │ │ │ │ # an IO error, so terminate the connection. │ 2706: │ 226 │ │ │ │ self._con.terminate() │ ... 2712: │ │ 2713: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 2714: │ │ budget = None │ │ 2715: │ │ self = <asyncpg.pool.PoolConnectionHolder object at │ │ 2716: │ │ 0x7ff38edb2c40> │ │ 2717: │ │ timeout = None │ │ 2718: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2719: │ │ 2720: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2721: │ ges/asyncpg/pool.py:218 in release │ 2722: │ │ 2723: │ 215 │ │ │ │ │ await self._con._reset() │ 2724: │ 216 │ │ │ │ │ await self._pool._reset(self._con) │ 2725: │ 217 │ │ │ else: │ 2726: │ ❱ 218 │ │ │ │ await self._con.reset(timeout=budget) │ 2727: │ 219 │ │ except (Exception, asyncio.CancelledError) as ex: │ 2728: │ 220 │ │ │ # If the `reset` call failed, terminate the connectio │ 2729: │ 221 │ │ │ # A new one will be created when `acquire` is called │ ... 2776: │ 350 │ │ │ return result │ 2777: │ 351 │ │ │ 2778: │ 352 │ │ _, status, _ = await self._execute( │ 2779: │ │ 2780: │ ╭────────────────────────────── locals ──────────────────────────────╮ │ 2781: │ │ args = () │ │ 2782: │ │ query = 'ROLLBACK' │ │ 2783: │ │ self = <asyncpg.connection.Connection object at 0x7ff3c23e1c70> │ │ 2784: │ │ timeout = None │ │ 2785: │ ╰────────────────────────────────────────────────────────────────────╯ │ 2786: │ │ 2787: │ in query:360 │ 2788: │ │ 2789: │ in asyncpg.protocol.protocol.BaseProtocol._check_state:745 │ 2790: ╰──────────────────────────────────────────────────────────────────────────╯ 2791: InterfaceError: cannot perform operation: another operation is in progress 2792: ───────────────────────── query: search docs by hybrid ───────────────────────── 2793: Failed at tests/test_docs_queries.py 2794: ╭─────────────────── Traceback (most recent call last) ────────────────────╮ 2795: │ in uvloop.loop.Loop.call_soon:1281 │ 2796: │ │ 2797: │ in uvloop.loop.Loop._call_soon:669 │ 2798: │ │ 2799: │ in uvloop.loop.UVStream.write:678 │ 2800: │ │ 2801: │ in uvloop.loop.Loop._append_ready_handle:673 │ 2802: │ │ 2803: │ in uvloop.loop.Loop._check_closed:705 │ 2804: ╰──────────────────────────────────────────────────────────────────────────╯ 2805: RuntimeError: Event loop is closed 2806: During handling of the above exception, another exception occurred: ... 2847: │ │ │ │ │ │ │ │ 2848: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2849: │ │ │ │ │ │ │ │ 2850: │ │ UUID('06828a99-8e7a-7a67-8000-f8aa992d91c6'), │ │ 2851: │ │ │ │ │ │ 'agent', │ │ 2852: │ │ │ │ │ │ │ │ 2853: │ │ UUID('06828a99-8b20-79c1-8000-002d13aee9b9') │ │ 2854: │ │ │ │ │ ], │ │ 2855: │ │ │ │ │ 'timeout': 90.0 │ │ 2856: │ │ │ │ } │ │ 2857: │ │ │ ) │ │ 2858: │ │ ] │ │ 2859: │ │ conn = <PoolConnectionProxy │ │ 2860: │ │ <asyncpg.connection.Connection object at │ │ 2861: │ │ 0x7ff3a3707c50> 0x7ff3a332ae90> │ │ 2862: │ │ connection_error = False │ │ 2863: │ │ connection_pool = <asyncpg.pool.Pool object at 0x7ff38c967ac0> │ │ 2864: │ │ debug = None │ │ 2865: │ │ kwargs = { │ │ 2866: │ │ │ 'developer_id': │ │ 2867: │ │ UUID('00000000-0000-0000-0000-000000000000'), │ │ 2868: │ │ │ 'doc_id': │ │ 2869: │ │ UUID('06828a99-8e7a-7a67-8000-f8aa992d91c6'), │ │ 2870: │ │ │ 'owner_type': 'agent', │ │ 2871: │ │ │ 'owner_id': │ │ 2872: │ │ UUID('06828a99-8b20-79c1-8000-002d13aee9b9') │ │ 2873: │ │ } │ │ 2874: │ │ only_on_error = False │ │ 2875: │ │ pool = <asyncpg.pool.Pool object at 0x7ff38c967ac0> │ │ ... 2898: │ │ │ │ │ │ │ 2899: │ │ UUID('06828a99-8e7a-7a67-8000-f8aa992d91c6'), │ │ 2900: │ │ │ │ │ 'agent', │ │ 2901: │ │ │ │ │ │ │ 2902: │ │ UUID('06828a99-8b20-79c1-8000-002d13aee9b9') │ │ 2903: │ │ │ │ ] │ │ 2904: │ │ │ ) │ │ 2905: │ │ ] │ │ 2906: │ │ return_index = -1 │ │ 2907: │ │ timeit = False │ │ 2908: │ ╰──────────────────────────────────────────────────────────────────────╯ │ 2909: │ │ 2910: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2911: │ ges/asyncpg/transaction.py:68 in __aenter__ │ 2912: │ │ 2913: │ 65 │ │ │ raise apg_errors.InterfaceError( │ 2914: │ 66 │ │ │ │ 'cannot enter context: already in an `async with` │ 2915: │ 67 │ │ self._managed = True │ 2916: │ ❱ 68 │ │ await self.start() │ 2917: │ 69 │ │ 2918: │ 70 │ async def __aexit__(self, extype, ex, tb): │ 2919: │ 71 │ │ try: │ 2920: │ │ 2921: │ ╭───────────────────────── locals ─────────────────────────╮ │ 2922: │ │ self = <asyncpg.Transaction state:failed 0x7ff3a048e0a0> │ │ 2923: │ ╰──────────────────────────────────────────────────────────╯ │ 2924: │ │ 2925: │ /home/runner/work/julep/julep/agents-api/.venv/lib/python3.12/site-packa │ 2926: │ ges/asyncpg/transaction.py:146 in start │ 2927: │ │ 2928: │ 143 │ │ │ query += ';' │ 2929: │ 144 │ │ │ 2930: │ 145 │ │ try: │ 2931: │ ❱ 146 │ │ │ await self._connection.execute(query) │ 2932: │ 147 │ │ except BaseException: │ 2933: │ 148 │ │ │ self._state = TransactionState.FAILED │ 2934: │ 149 │ │ │ raise │ 2935: │ │ 2936: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 2937: │ │ con = <asyncpg.connection.Connection object at 0x7ff3a3707c50> │ │ 2938: │ │ query = 'BEGIN;' │ │ 2939: │ │ self = <asyncpg.Transaction state:failed 0x7ff3a048e0a0> │ │ 2940: │ ╰──────────────────────────────────────────────────────────────────╯ │ ... 2957: │ │ timeout = None │ │ 2958: │ ╰────────────────────────────────────────────────────────────────────╯ │ 2959: │ │ 2960: │ in query:375 │ 2961: │ │ 2962: │ in asyncpg.protocol.protocol.BaseProtocol.query:368 │ 2963: │ │ 2964: │ in asyncpg.protocol.protocol.CoreProtocol._simple_query:1174 │ 2965: │ │ 2966: │ in asyncpg.protocol.protocol.BaseProtocol._write:967 │ 2967: │ │ 2968: │ in uvloop.loop.UVStream.write:678 │ 2969: │ │ 2970: │ in uvloop.loop.UVHandle._ensure_alive:159 │ 2971: ╰──────────────────────────────────────────────────────────────────────────╯ 2972: RuntimeError: unable to perform operation on <TCPTransport closed=True 2973: reading=False 0x4163c920>; the handler is closed ... 2986: │ │ 2987: │ ╭───────────────────────────── locals ─────────────────────────────╮ │ 2988: │ │ debug = None │ │ 2989: │ │ loop_factory = None │ │ 2990: │ │ main = <async_generator_asend object at 0x7ff3a341fc00> │ │ 2991: │ │ runner = <asyncio.runners.Runner object at 0x7ff3a3328410> │ │ 2992: │ ╰──────────────────────────────────────────────────────────────────╯ │ 2993: │ │ 2994: │ /home/runner/.local/share/uv/python/cpython-3.12.10-linux-x86_64-gnu/lib │ 2995: │ /python3.12/asyncio/runners.py:118 in run │ 2996: │ │ 2997: │ 115 │ │ │ 2998: │ 116 │ │ self._interrupt_count = 0 │ 2999: │ 117 │ │ try: │ 3000: │ ❱ 118 │ │ │ return self._loop.run_until_complete(task) │ 3001: │ 119 │ │ except exceptions.CancelledError: │ 3002: │ 120 │ │ │ if self._interrupt_count > 0: │ 3003: │ 121 │ │ │ │ uncancel = getattr(task, "uncancel", None) │ 3004: │ │ 3005: │ ╭─────────────────────────────── locals ───────────────────────────────╮ │ 3006: │ │ context = <_contextvars.Context object at 0x7ff3a3337400> │ │ 3007: │ │ coro = <async_generator_asend object at 0x7ff3a341fc00> │ │ 3008: │ │ self = <asyncio.runners.Runner object at 0x7ff3a3328410> │ │ 3009: │ │ sigint_handler = functools.partial(<bound method Runner._on_sigint │ │ 3010: │ │ of <asyncio.runners.Runner object at │ │ 3011: │ │ 0x7ff3a3328410>>, main_task=<Task finished │ │ 3012: │ │ name='Task-2168' coro=<<async_generator_asend │ │ 3013: │ │ without __name__>()> │ │ 3014: │ │ exception=InterfaceError('cannot perform operation: │ │ 3015: │ │ another operation is in progress')>) │ │ 3016: │ │ task = <Task finished name='Task-2168' │ │ 3017: │ │ coro=<<async_generator_asend without __name__>()> │ │ 3018: │ │ exception=InterfaceError('cannot perform operation: │ │ 3019: │ │ another operation is in progress')> │ │ ... 3069: │ │ 905090, tzinfo=datetime.timezone.utc), │ │ 3070: │ │ │ title='Hello', │ │ 3071: │ │ │ content=['World', 'World2', 'World3'], │ │ 3072: │ │ │ embeddings=[], │ │ 3073: │ │ │ modality='text', │ ...

[qodo-merge-for-open-source[bot]]

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 1 🔵⚪⚪⚪⚪

🧪 No relevant tests

🔒 No security concerns identified

⚡ No major issues detected

[qodo-merge-for-open-source[bot]]

PR Code Suggestions ✨

No code suggestions found for the PR.