Harrison/arxiv fix (#5993)

Co-authored-by: Juanjo do Olmo <87780148+SimplyJuanjo@users.noreply.github.com>
This commit is contained in:
Harrison Chase 2023-06-11 09:48:09 -07:00 committed by GitHub
parent 0b740c9baa
commit 2d038b57b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,6 +2,8 @@
from typing import Optional
from pydantic import Field
from langchain.callbacks.manager import (
AsyncCallbackManagerForToolRun,
CallbackManagerForToolRun,
@ -22,7 +24,7 @@ class ArxivQueryRun(BaseTool):
"from scientific articles on arxiv.org. "
"Input should be a search query."
)
api_wrapper: ArxivAPIWrapper
api_wrapper: ArxivAPIWrapper = Field(default_factory=ArxivAPIWrapper)
def _run(
self,