mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +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
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
import json
|
|
|
|
import pytest
|
|
import responses
|
|
|
|
from langchain_community.utilities.graphql import GraphQLAPIWrapper
|
|
|
|
TEST_ENDPOINT = "http://testserver/graphql"
|
|
|
|
# Mock GraphQL response for testing
|
|
MOCK_RESPONSE = {
|
|
"data": {
|
|
"allUsers": [{"name": "Alice"}],
|
|
"__schema": {
|
|
"queryType": {"name": "Query"},
|
|
"types": [
|
|
{
|
|
"kind": "OBJECT",
|
|
"name": "Query",
|
|
"fields": [
|
|
{
|
|
"name": "allUsers",
|
|
"args": [],
|
|
"type": {
|
|
"kind": "NON_NULL",
|
|
"name": None,
|
|
"ofType": {
|
|
"kind": "OBJECT",
|
|
"name": "allUsers",
|
|
"ofType": None,
|
|
},
|
|
},
|
|
}
|
|
],
|
|
"inputFields": None,
|
|
"interfaces": [],
|
|
"enumValues": None,
|
|
"possibleTypes": None,
|
|
},
|
|
{
|
|
"kind": "SCALAR",
|
|
"name": "String",
|
|
},
|
|
{
|
|
"kind": "OBJECT",
|
|
"name": "allUsers",
|
|
"description": None,
|
|
"fields": [
|
|
{
|
|
"name": "name",
|
|
"description": None,
|
|
"args": [],
|
|
"type": {
|
|
"kind": "NON_NULL",
|
|
"name": None,
|
|
"ofType": {
|
|
"kind": "SCALAR",
|
|
"name": "String",
|
|
"ofType": None,
|
|
},
|
|
},
|
|
},
|
|
],
|
|
"inputFields": None,
|
|
"interfaces": [],
|
|
"enumValues": None,
|
|
"possibleTypes": None,
|
|
},
|
|
{
|
|
"kind": "SCALAR",
|
|
"name": "Boolean",
|
|
},
|
|
],
|
|
},
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.mark.requires("gql", "requests_toolbelt")
|
|
@responses.activate
|
|
def test_run() -> None:
|
|
responses.add(responses.POST, TEST_ENDPOINT, json=MOCK_RESPONSE, status=200)
|
|
|
|
query = "query { allUsers { name } }"
|
|
graphql_wrapper = GraphQLAPIWrapper(
|
|
graphql_endpoint=TEST_ENDPOINT,
|
|
custom_headers={"Authorization": "Bearer testtoken"},
|
|
)
|
|
result = graphql_wrapper.run(query)
|
|
|
|
expected_result = json.dumps(MOCK_RESPONSE["data"], indent=2)
|
|
assert result == expected_result
|