googleapis / googleapis/python-aiplatform
Bug: `async_stream_query` not async
- 主要语言
- Python
- 星标
- 905
- 派生
- 465
- 平均合并
- 1 天 13 小时
- 30 天内合并 PR
- 44
描述
#### Environment details
- OS type and version: Linux 6.6.87.2-microsoft-standard-WSL2
- Python version: Python 3.11.9
- pip version: pip 25.2
- `google-cloud-aiplatform` version: 1.111.0
#### Steps to reproduce
1. Deploy a simple adk agent.
2. Run multiple async functions to call the agent's `async_stream_query` method.
#### Code example
```python
async def run_query(app, uid):
prompt = 'some prompts have delayed responses'
session = await app.async_create_session(user_id=uid)
async for event in app.async_stream_query(
user_id=uid,
session_id=session['id'],
message=prompt,
):
print(event)
await app.async_delete_session(user_id=uid, session_id=session['id'])
async def run():
agent = agent_engines.get(AGENT_ID)
await asyncio.gather(
run_query(agent, 'u1'),
run_query(agent, 'u2'),
run_query(agent, 'u3'),
)
if __name__ == '__main__':
asyncio.run(run())
```
#### Result
The requests are not sent concurrently to the Vertex AI Agent Engine.
贡献指南
调研方向
首先定位 async_stream_query 的实现,并检查它在复现代码中如何与 async_create_session、async_delete_session 和 asyncio.gather 一起使用。使用多个延迟的提示来验证行为,并在并发调用确实同时发送到 Vertex AI Agent Engine 时认为该 issue 已完成。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- google-cloud, python
- 领域
- backend-api-design, machine-learning
- Issue 类型
- 缺陷
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 活跃度
- 冷清
- 描述清晰度
- 基本清楚
- 新手友好度
- 45/100