mirror of
https://github.com/hwchase17/langchain
synced 2024-11-04 06:00:26 +00:00
ed58eeb9c5
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
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Optional
|
|
|
|
from langchain_core.callbacks import CallbackManagerForToolRun
|
|
|
|
from langchain_community.tools.edenai.edenai_base_tool import EdenaiTool
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class EdenAiExplicitImageTool(EdenaiTool):
|
|
|
|
"""Tool that queries the Eden AI Explicit image detection.
|
|
|
|
for api reference check edenai documentation:
|
|
https://docs.edenai.co/reference/image_explicit_content_create.
|
|
|
|
To use, you should have
|
|
the environment variable ``EDENAI_API_KEY`` set with your API token.
|
|
You can find your token here: https://app.edenai.run/admin/account/settings
|
|
|
|
"""
|
|
|
|
name = "edenai_image_explicit_content_detection"
|
|
|
|
description = (
|
|
"A wrapper around edenai Services Explicit image detection. "
|
|
"""Useful for when you have to extract Explicit Content from images.
|
|
it detects adult only content in images,
|
|
that is generally inappropriate for people under
|
|
the age of 18 and includes nudity, sexual activity,
|
|
pornography, violence, gore content, etc."""
|
|
"Input should be the string url of the image ."
|
|
)
|
|
|
|
combine_available = True
|
|
feature = "image"
|
|
subfeature = "explicit_content"
|
|
|
|
def _parse_json(self, json_data: dict) -> str:
|
|
result_str = f"nsfw_likelihood: {json_data['nsfw_likelihood']}\n"
|
|
for idx, found_obj in enumerate(json_data["items"]):
|
|
label = found_obj["label"].lower()
|
|
likelihood = found_obj["likelihood"]
|
|
result_str += f"{idx}: {label} likelihood {likelihood},\n"
|
|
|
|
return result_str[:-2]
|
|
|
|
def _parse_response(self, json_data: list) -> str:
|
|
if len(json_data) == 1:
|
|
result = self._parse_json(json_data[0])
|
|
else:
|
|
for entry in json_data:
|
|
if entry.get("provider") == "eden-ai":
|
|
result = self._parse_json(entry)
|
|
|
|
return result
|
|
|
|
def _run(
|
|
self,
|
|
query: str,
|
|
run_manager: Optional[CallbackManagerForToolRun] = None,
|
|
) -> str:
|
|
"""Use the tool."""
|
|
query_params = {"file_url": query, "attributes_as_list": False}
|
|
return self._call_eden_ai(query_params)
|