mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-07 22:11:51 +00:00
community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463)
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
This commit is contained in:
133
libs/community/langchain_community/tools/json/tool.py
Normal file
133
libs/community/langchain_community/tools/json/tool.py
Normal file
@@ -0,0 +1,133 @@
|
||||
# flake8: noqa
|
||||
"""Tools for working with JSON specs."""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import re
|
||||
from pathlib import Path
|
||||
from typing import Dict, List, Optional, Union
|
||||
|
||||
from langchain_core.pydantic_v1 import BaseModel
|
||||
|
||||
from langchain_core.callbacks import (
|
||||
AsyncCallbackManagerForToolRun,
|
||||
CallbackManagerForToolRun,
|
||||
)
|
||||
from langchain_core.tools import BaseTool
|
||||
|
||||
|
||||
def _parse_input(text: str) -> List[Union[str, int]]:
|
||||
"""Parse input of the form data["key1"][0]["key2"] into a list of keys."""
|
||||
_res = re.findall(r"\[.*?]", text)
|
||||
# strip the brackets and quotes, convert to int if possible
|
||||
res = [i[1:-1].replace('"', "").replace("'", "") for i in _res]
|
||||
res = [int(i) if i.isdigit() else i for i in res]
|
||||
return res
|
||||
|
||||
|
||||
class JsonSpec(BaseModel):
|
||||
"""Base class for JSON spec."""
|
||||
|
||||
dict_: Dict
|
||||
max_value_length: int = 200
|
||||
|
||||
@classmethod
|
||||
def from_file(cls, path: Path) -> JsonSpec:
|
||||
"""Create a JsonSpec from a file."""
|
||||
if not path.exists():
|
||||
raise FileNotFoundError(f"File not found: {path}")
|
||||
dict_ = json.loads(path.read_text())
|
||||
return cls(dict_=dict_)
|
||||
|
||||
def keys(self, text: str) -> str:
|
||||
"""Return the keys of the dict at the given path.
|
||||
|
||||
Args:
|
||||
text: Python representation of the path to the dict (e.g. data["key1"][0]["key2"]).
|
||||
"""
|
||||
try:
|
||||
items = _parse_input(text)
|
||||
val = self.dict_
|
||||
for i in items:
|
||||
if i:
|
||||
val = val[i]
|
||||
if not isinstance(val, dict):
|
||||
raise ValueError(
|
||||
f"Value at path `{text}` is not a dict, get the value directly."
|
||||
)
|
||||
return str(list(val.keys()))
|
||||
except Exception as e:
|
||||
return repr(e)
|
||||
|
||||
def value(self, text: str) -> str:
|
||||
"""Return the value of the dict at the given path.
|
||||
|
||||
Args:
|
||||
text: Python representation of the path to the dict (e.g. data["key1"][0]["key2"]).
|
||||
"""
|
||||
try:
|
||||
items = _parse_input(text)
|
||||
val = self.dict_
|
||||
for i in items:
|
||||
val = val[i]
|
||||
|
||||
if isinstance(val, dict) and len(str(val)) > self.max_value_length:
|
||||
return "Value is a large dictionary, should explore its keys directly"
|
||||
str_val = str(val)
|
||||
if len(str_val) > self.max_value_length:
|
||||
str_val = str_val[: self.max_value_length] + "..."
|
||||
return str_val
|
||||
except Exception as e:
|
||||
return repr(e)
|
||||
|
||||
|
||||
class JsonListKeysTool(BaseTool):
|
||||
"""Tool for listing keys in a JSON spec."""
|
||||
|
||||
name: str = "json_spec_list_keys"
|
||||
description: str = """
|
||||
Can be used to list all keys at a given path.
|
||||
Before calling this you should be SURE that the path to this exists.
|
||||
The input is a text representation of the path to the dict in Python syntax (e.g. data["key1"][0]["key2"]).
|
||||
"""
|
||||
spec: JsonSpec
|
||||
|
||||
def _run(
|
||||
self,
|
||||
tool_input: str,
|
||||
run_manager: Optional[CallbackManagerForToolRun] = None,
|
||||
) -> str:
|
||||
return self.spec.keys(tool_input)
|
||||
|
||||
async def _arun(
|
||||
self,
|
||||
tool_input: str,
|
||||
run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
|
||||
) -> str:
|
||||
return self._run(tool_input)
|
||||
|
||||
|
||||
class JsonGetValueTool(BaseTool):
|
||||
"""Tool for getting a value in a JSON spec."""
|
||||
|
||||
name: str = "json_spec_get_value"
|
||||
description: str = """
|
||||
Can be used to see value in string format at a given path.
|
||||
Before calling this you should be SURE that the path to this exists.
|
||||
The input is a text representation of the path to the dict in Python syntax (e.g. data["key1"][0]["key2"]).
|
||||
"""
|
||||
spec: JsonSpec
|
||||
|
||||
def _run(
|
||||
self,
|
||||
tool_input: str,
|
||||
run_manager: Optional[CallbackManagerForToolRun] = None,
|
||||
) -> str:
|
||||
return self.spec.value(tool_input)
|
||||
|
||||
async def _arun(
|
||||
self,
|
||||
tool_input: str,
|
||||
run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
|
||||
) -> str:
|
||||
return self._run(tool_input)
|
Reference in New Issue
Block a user