File tree Expand file tree Collapse file tree
tests/unit/experimental/components Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3333)
3434from neo4j_graphrag .experimental .pipeline .exceptions import InvalidJSONError
3535from neo4j_graphrag .llm import LLMInterface , LLMResponse
36- from neo4j_graphrag .llm import AnthropicLLM , OpenAILLM , VertexAILLM
36+ from neo4j_graphrag .llm import OpenAILLM , VertexAILLM
3737from unittest .mock import patch
3838
3939
@@ -461,7 +461,8 @@ async def test_extractor_structured_output_with_vertexai() -> None:
461461
462462def test_extractor_structured_output_unsupported_llm () -> None :
463463 """Test that use_structured_output=True raises error with unsupported LLMs."""
464- llm = AnthropicLLM (api_key = "test" , model_name = "claude-3-opus" )
464+ llm = MagicMock (spec = LLMInterface )
465+ llm .supports_structured_output = False
465466
466467 with pytest .raises (ValueError ) as exc_info :
467468 LLMEntityRelationExtractor (
You can’t perform that action at this time.
0 commit comments