googleapis / googleapis/python-aiplatform

Missing type stub for agent_engines.memories.retrieve() method

未关闭
#6,166 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
api: vertex-ai
主要语言
Python
星标
905
派生
465
平均合并
1 天 13 小时
30 天内合并 PR
44

描述

# Missing Type Stubs for `client.agent_engines.memories.retrieve()` in Google ADK

## **Description**

The `client.agent_engines.memories.retrieve()` method works correctly at runtime, but it is **missing from the type stubs**. As a result, IDEs and type checkers (Pylance, Pyright, mypy) report errors such as:

```
Cannot access attribute "retrieve" for class "_AgentEngineMemories"
```

The method itself *does* exist and functions properly during execution.

---

## **Environment**

* **google-adk version:** (run: `pip show google-adk`)
* **vertexai version:** (run: `pip show google-cloud-aiplatform`)
* **Python version:** 3.12

---

## **Steps to Reproduce**

```python
import vertexai

client = vertexai.Client(project="my-project", location="us-central1")

# Works at runtime, but IDE reports a type error
retrieved_memories = client.agent_engines.memories.retrieve(
name="projects/.../reasoningEngines/...",
scope={"app_name": "test", "user_id": "user123"},
similarity_search_params={"search_query": "*"}
)
```

---

## **Expected Behavior**

The SDK's type stubs should correctly declare the `retrieve()` method so that IDEs and type checkers do not show false errors.

---

## **Actual Behavior**

* Runtime: **works correctly**, method is available and returns results.
* Type checking: fails with errors such as:

* `Cannot access attribute "retrieve" for class "_AgentEngineMemories"`

---

## **Additional Context**

* The implementation exists in `vertexai/_genai/memories.py` (around lines 1182–1233, inside the `Memories` class).
* However, the **type stub files (.pyi)** do not expose this method.
* Other methods under `agent_engines.memories` such as `generate()`, `create()`, `list()`, and `get()` may also require type stub validation.

This issue affects static analysis and IDE tooling but not runtime functionality. A stub update should resolve it.

贡献指南

打开贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。