mirror of
https://github.com/hwchase17/langchain.git
synced 2025-05-04 22:58:42 +00:00
600b7bdd61
Co-authored-by: Bagatur <baskaryan@gmail.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from langchain_core.callbacks import CallbackManagerForToolRun
|
|
from langchain_core.tools import BaseTool
|
|
from pydantic import BaseModel, Field
|
|
|
|
from langchain_community.utilities.jina_search import JinaSearchAPIWrapper
|
|
|
|
|
|
class JinaInput(BaseModel):
|
|
"""Input for the Jina search tool."""
|
|
|
|
query: str = Field(description="search query to look up")
|
|
|
|
|
|
class JinaSearch(BaseTool): # type: ignore[override]
|
|
"""Tool that queries the JinaSearch.
|
|
|
|
..versionadded:: 0.2.16
|
|
"""
|
|
|
|
name: str = "jina_search"
|
|
description: str = (
|
|
"Jina Reader allows you to ground your LLM with the latest information from "
|
|
"the web. "
|
|
"Jina Reader will search the web and return the top five results with their "
|
|
"URLs and contents, "
|
|
"each in clean, LLM-friendly text. This way, you can always keep your LLM "
|
|
"up-to-date, improve its factuality, and reduce hallucinations."
|
|
)
|
|
search_wrapper: JinaSearchAPIWrapper = Field(default_factory=JinaSearchAPIWrapper)
|
|
|
|
def _run(
|
|
self,
|
|
query: str,
|
|
run_manager: Optional[CallbackManagerForToolRun] = None,
|
|
) -> str:
|
|
"""Use the tool."""
|
|
return self.search_wrapper.run(query)
|