mirror of
https://github.com/langgenius/dify.git
synced 2026-05-05 09:58:04 +08:00
refactor(api): rename dify_graph to graphon (#34095)
This commit is contained in:
@ -5,8 +5,8 @@ from core.rag.data_post_processor.reorder import ReorderRunner
|
||||
from core.rag.index_processor.constant.query_type import QueryType
|
||||
from core.rag.models.document import Document
|
||||
from core.rag.rerank.rerank_type import RerankMode
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelType
|
||||
from dify_graph.model_runtime.errors.invoke import InvokeAuthorizationError
|
||||
from graphon.model_runtime.entities.model_entities import ModelType
|
||||
from graphon.model_runtime.errors.invoke import InvokeAuthorizationError
|
||||
|
||||
|
||||
def _doc(content: str) -> Document:
|
||||
|
||||
@ -15,8 +15,8 @@ import pytest
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
|
||||
from core.rag.embedding.cached_embedding import CacheEmbedding
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelPropertyKey
|
||||
from dify_graph.model_runtime.entities.text_embedding_entities import EmbeddingResult, EmbeddingUsage
|
||||
from graphon.model_runtime.entities.model_entities import ModelPropertyKey
|
||||
from graphon.model_runtime.entities.text_embedding_entities import EmbeddingResult, EmbeddingUsage
|
||||
from models.dataset import Embedding
|
||||
|
||||
|
||||
|
||||
@ -53,9 +53,9 @@ from sqlalchemy.exc import IntegrityError
|
||||
|
||||
from core.entities.embedding_type import EmbeddingInputType
|
||||
from core.rag.embedding.cached_embedding import CacheEmbedding
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelPropertyKey
|
||||
from dify_graph.model_runtime.entities.text_embedding_entities import EmbeddingResult, EmbeddingUsage
|
||||
from dify_graph.model_runtime.errors.invoke import (
|
||||
from graphon.model_runtime.entities.model_entities import ModelPropertyKey
|
||||
from graphon.model_runtime.entities.text_embedding_entities import EmbeddingResult, EmbeddingUsage
|
||||
from graphon.model_runtime.errors.invoke import (
|
||||
InvokeAuthorizationError,
|
||||
InvokeConnectionError,
|
||||
InvokeRateLimitError,
|
||||
|
||||
@ -7,9 +7,9 @@ from core.entities.knowledge_entities import PreviewDetail
|
||||
from core.rag.index_processor.constant.index_type import IndexTechniqueType
|
||||
from core.rag.index_processor.processor.paragraph_index_processor import ParagraphIndexProcessor
|
||||
from core.rag.models.document import AttachmentDocument, Document
|
||||
from dify_graph.model_runtime.entities.llm_entities import LLMResult, LLMUsage
|
||||
from dify_graph.model_runtime.entities.message_entities import AssistantPromptMessage, ImagePromptMessageContent
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelFeature
|
||||
from graphon.model_runtime.entities.llm_entities import LLMResult, LLMUsage
|
||||
from graphon.model_runtime.entities.message_entities import AssistantPromptMessage, ImagePromptMessageContent
|
||||
from graphon.model_runtime.entities.model_entities import ModelFeature
|
||||
|
||||
|
||||
class TestParagraphIndexProcessor:
|
||||
|
||||
@ -63,7 +63,7 @@ from core.indexing_runner import (
|
||||
)
|
||||
from core.rag.index_processor.constant.index_type import IndexStructureType, IndexTechniqueType
|
||||
from core.rag.models.document import ChildDocument, Document
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelType
|
||||
from graphon.model_runtime.entities.model_entities import ModelType
|
||||
from libs.datetime_utils import naive_utc_now
|
||||
from models.dataset import Dataset, DatasetProcessRule
|
||||
from models.dataset import Document as DatasetDocument
|
||||
|
||||
@ -28,7 +28,7 @@ from core.rag.rerank.rerank_factory import RerankRunnerFactory
|
||||
from core.rag.rerank.rerank_model import RerankModelRunner
|
||||
from core.rag.rerank.rerank_type import RerankMode
|
||||
from core.rag.rerank.weight_rerank import WeightRerankRunner
|
||||
from dify_graph.model_runtime.entities.rerank_entities import RerankDocument, RerankResult
|
||||
from graphon.model_runtime.entities.rerank_entities import RerankDocument, RerankResult
|
||||
|
||||
|
||||
def create_mock_model_instance() -> ModelInstance:
|
||||
|
||||
@ -35,8 +35,8 @@ from core.rag.retrieval.dataset_retrieval import DatasetRetrieval
|
||||
from core.rag.retrieval.retrieval_methods import RetrievalMethod
|
||||
from core.workflow.nodes.knowledge_retrieval import exc
|
||||
from core.workflow.nodes.knowledge_retrieval.retrieval import KnowledgeRetrievalRequest
|
||||
from dify_graph.model_runtime.entities.llm_entities import LLMUsage
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelFeature
|
||||
from graphon.model_runtime.entities.llm_entities import LLMUsage
|
||||
from graphon.model_runtime.entities.model_entities import ModelFeature
|
||||
from models.dataset import Dataset
|
||||
from models.enums import CreatorUserRole
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
from unittest.mock import Mock
|
||||
|
||||
from core.rag.retrieval.router.multi_dataset_function_call_router import FunctionCallMultiDatasetRouter
|
||||
from dify_graph.model_runtime.entities.llm_entities import LLMUsage
|
||||
from graphon.model_runtime.entities.llm_entities import LLMUsage
|
||||
|
||||
|
||||
class TestFunctionCallMultiDatasetRouter:
|
||||
|
||||
@ -3,9 +3,9 @@ from unittest.mock import Mock, patch
|
||||
|
||||
from core.rag.retrieval.output_parser.react_output import ReactAction, ReactFinish
|
||||
from core.rag.retrieval.router.multi_dataset_react_route import ReactMultiDatasetRouter
|
||||
from dify_graph.model_runtime.entities.llm_entities import LLMUsage
|
||||
from dify_graph.model_runtime.entities.message_entities import PromptMessageRole
|
||||
from dify_graph.model_runtime.entities.model_entities import ModelType
|
||||
from graphon.model_runtime.entities.llm_entities import LLMUsage
|
||||
from graphon.model_runtime.entities.message_entities import PromptMessageRole
|
||||
from graphon.model_runtime.entities.model_entities import ModelType
|
||||
|
||||
|
||||
class TestReactMultiDatasetRouter:
|
||||
|
||||
Reference in New Issue
Block a user