googleapis / googleapis/python-aiplatform

[Reasoning Engine] LangchainAgent.stream_query attempts to import from deprecated langchain.load path

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

描述

#### Environment details

- OS type and version: Ubuntu 24.04.4 LTS
- Python version: `3.12.10`
- pip version: `25.2`
- `google-cloud-aiplatform` version: `Latest from main`

#### Steps to reproduce

1. Initialize a `LangchainAgent` or `LanggraphAgent` from the agent engine or reasoning engine templates.
2. Call the `query` or `stream_query` method.
3. Observe that it fails with an `ImportError` because it attempts to import from `langchain.load` instead of `langchain_core.load`.

#### Code example

```python
from vertexai.agent_engines.templates.langchain import LangchainAgent

# Initialize agent
agent = LangchainAgent(model="gemini-1.5-flash-001")
agent.set_up()

# Both query and stream_query are affected
result = agent.query(input="Hello")

for chunk in agent.stream_query(input="Hello"):
print(chunk)
```

#### Stack trace
```
Traceback (most recent call last):
File "example.py", line 8, in
for chunk in agent.stream_query(input="Hello"):
File ".../vertexai/agent_engines/templates/langchain.py", line 665, in stream_query
from langchain.load import dump as langchain_load_dump
ImportError: cannot import name 'dump' from 'langchain.load'
```

#### Affected files

The same deprecated import (`from langchain.load import dump`) exists in both `query()` and `stream_query()` across all 4 template modules:

- `vertexai/agent_engines/templates/langchain.py`
- `vertexai/agent_engines/templates/langgraph.py`
- `vertexai/preview/reasoning_engines/templates/langchain.py`
- `vertexai/preview/reasoning_engines/templates/langgraph.py`

Making sure to follow these steps will guarantee the quickest resolution possible.

Thanks!

贡献指南

打开贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

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