youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-30 02:20:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api): use standalone graphon package (#34209)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
99
2026-03-28 05:05:32 +08:00
committed by GitHub
parent 32d394d65b
commit 40591a7c50
883 changed files with 1779 additions and 47377 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/tests/unit_tests/utils/structured_output_parser/test_structured_output_parser.py
View File

@ -2,9 +2,6 @@ from decimal import Decimal
from unittest.mock import MagicMock, patch
import pytest
from core.llm_generator.output_parser.errors import OutputParserError
from core.llm_generator.output_parser.structured_output import invoke_llm_with_structured_output
from graphon.model_runtime.entities.llm_entities import (
LLMResult,
LLMResultChunk,
@ -21,6 +18,9 @@ from graphon.model_runtime.entities.message_entities import (
)
from graphon.model_runtime.entities.model_entities import AIModelEntity, ModelType
from core.llm_generator.output_parser.errors import OutputParserError
from core.llm_generator.output_parser.structured_output import invoke_llm_with_structured_output
def create_mock_usage(prompt_tokens: int = 10, completion_tokens: int = 5) -> LLMUsage:
"""Create a mock LLMUsage with all required fields"""