github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-28 06:48:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix more issues (#26116)

Browse Source 
This PR resolves more type checking issues and fixes some bugs.
This commit is contained in:
Eugene Yurtsev
2024-09-05 16:31:21 -04:00
committed by GitHub
parent 0cc6584889
commit e02b093d81
12 changed files with 43 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/tests/unit_tests/chat_models/test_oci_generative_ai.py
View File

@@ -23,7 +23,11 @@ def test_llm_chat(monkeypatch: MonkeyPatch, test_model_id: str) -> None:
oci_gen_ai_client = MagicMock()
llm = ChatOCIGenAI(model_id=test_model_id, client=oci_gen_ai_client)
provider = llm.model_id.split(".")[0].lower()
model_id = llm.model_id
if model_id is None:
raise ValueError("Model ID is required for OCI Generative AI LLM service.")
provider = model_id.split(".")[0].lower()
def mocked_response(*args): # type: ignore[no-untyped-def]
response_text = "Assistant chat reply."

10
libs/community/tests/unit_tests/document_loaders/test_onenote.py
View File

@@ -15,7 +15,6 @@ def test_initialization() -> None:
os.environ["MS_GRAPH_CLIENT_SECRET"] = "CLIENT_SECRET"
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
section_name="test_section",
page_title="test_title",
@@ -40,7 +39,6 @@ def test_load(mocker: MockerFixture) -> None:
return_value=mocker.MagicMock(json=lambda: {"value": []}, links=None),
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
section_name="test_section",
page_title="test_title",
@@ -57,7 +55,6 @@ def test_load(mocker: MockerFixture) -> None:
),
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
object_ids=["test_id"],
access_token="access_token",
)
@@ -85,7 +82,6 @@ def test_msal_import(monkeypatch: MonkeyPatch, mocker: MockerFixture) -> None:
return_value=FakeConfidentialClientApplication(),
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
section_name="test_section",
page_title="test_title",
@@ -99,7 +95,6 @@ def test_url() -> None:
os.environ["MS_GRAPH_CLIENT_SECRET"] = "CLIENT_SECRET"
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
section_name="test_section",
page_title="test_title",
@@ -115,7 +110,6 @@ def test_url() -> None:
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
section_name="test_section",
access_token="access_token",
@@ -129,7 +123,6 @@ def test_url() -> None:
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
notebook_name="test_notebook",
access_token="access_token",
onenote_api_base_url="https://graph.microsoft.com/v1.0/me/onenote",
@@ -141,7 +134,6 @@ def test_url() -> None:
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
section_name="test_section",
access_token="access_token",
onenote_api_base_url="https://graph.microsoft.com/v1.0/me/onenote",
@@ -153,7 +145,6 @@ def test_url() -> None:
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
section_name="test_section",
page_title="test_title",
access_token="access_token",
@@ -167,7 +158,6 @@ def test_url() -> None:
)
loader = OneNoteLoader(
settings={"client_id": "", "client_secret": ""},
page_title="test_title",
access_token="access_token",
onenote_api_base_url="https://graph.microsoft.com/v1.0/me/onenote",

7
libs/community/tests/unit_tests/llms/test_oci_generative_ai.py
View File

@@ -22,7 +22,12 @@ def test_llm_complete(monkeypatch: MonkeyPatch, test_model_id: str) -> None:
oci_gen_ai_client = MagicMock()
llm = OCIGenAI(model_id=test_model_id, client=oci_gen_ai_client)
provider = llm.model_id.split(".")[0].lower()
model_id = llm.model_id
if model_id is None:
raise ValueError("Model ID is required for OCI Generative AI LLM service.")
provider = model_id.split(".")[0].lower()
def mocked_response(*args): # type: ignore[no-untyped-def]
response_text = "This is the completion."