mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +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
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import json
|
|
import os
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from langchain_community.chat_models.azure_openai import AzureChatOpenAI
|
|
|
|
|
|
@mock.patch.dict(
|
|
os.environ,
|
|
{
|
|
"OPENAI_API_KEY": "test",
|
|
"OPENAI_API_BASE": "https://oai.azure.com/",
|
|
"OPENAI_API_VERSION": "2023-05-01",
|
|
},
|
|
)
|
|
@pytest.mark.requires("openai")
|
|
@pytest.mark.parametrize(
|
|
"model_name", ["gpt-4", "gpt-4-32k", "gpt-35-turbo", "gpt-35-turbo-16k"]
|
|
)
|
|
def test_model_name_set_on_chat_result_when_present_in_response(
|
|
model_name: str,
|
|
) -> None:
|
|
sample_response_text = f"""
|
|
{{
|
|
"id": "chatcmpl-7ryweq7yc8463fas879t9hdkkdf",
|
|
"object": "chat.completion",
|
|
"created": 1690381189,
|
|
"model": "{model_name}",
|
|
"choices": [
|
|
{{
|
|
"index": 0,
|
|
"finish_reason": "stop",
|
|
"message": {{
|
|
"role": "assistant",
|
|
"content": "I'm an AI assistant that can help you."
|
|
}}
|
|
}}
|
|
],
|
|
"usage": {{
|
|
"completion_tokens": 28,
|
|
"prompt_tokens": 15,
|
|
"total_tokens": 43
|
|
}}
|
|
}}
|
|
"""
|
|
# convert sample_response_text to instance of Mapping[str, Any]
|
|
sample_response = json.loads(sample_response_text)
|
|
mock_chat = AzureChatOpenAI()
|
|
chat_result = mock_chat._create_chat_result(sample_response)
|
|
assert (
|
|
chat_result.llm_output is not None
|
|
and chat_result.llm_output["model_name"] == model_name
|
|
)
|