Skip to content

Commit

Permalink
Remove unfinished LangChain Azure OpenAI test.
Browse files Browse the repository at this point in the history
  • Loading branch information
rmitsch committed Oct 5, 2023
1 parent 55f1db0 commit dce80b0
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 20 deletions.
19 changes: 0 additions & 19 deletions spacy_llm/tests/models/test_langchain.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,22 +21,3 @@ def test_initialization():
nlp = spacy.blank("en")
nlp.add_pipe("llm", config=PIPE_CFG)
nlp("This is a test.")


@pytest.mark.external
@pytest.mark.skipif(has_langchain is False, reason="LangChain is not installed")
def test_initialization_azure_openai():
"""Test initialization and simple run with Azure OpenAI models."""
pipe_cfg = {
"model": {
"@llm_models": "langchain.Azure.v1",
"name": "ada",
"config": {"temperature": 0.3},
"query": {"@llm_queries": "spacy.CallLangChain.v1"},
},
"task": {"@llm_tasks": "spacy.NoOp.v1"},
}

nlp = spacy.blank("en")
nlp.add_pipe("llm", config=pipe_cfg)
nlp("This is a test.")
2 changes: 1 addition & 1 deletion spacy_llm/tests/models/test_rest.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ def test_max_time_error_handling():


@pytest.mark.skipif(
has_azure_openai_key is False, reason="OpenAI API key not available"
has_azure_openai_key is False, reason="Azure OpenAI API key not available"
)
@pytest.mark.external
@pytest.mark.parametrize("deployment_name", ("gpt-35-turbo", "gpt-35-turbo-instruct"))
Expand Down

0 comments on commit dce80b0

Please sign in to comment.