models.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. from shared.abstractions import (
  2. AggregateSearchResult,
  3. ChunkSearchResult,
  4. GenerationConfig,
  5. GraphCommunityResult,
  6. GraphEntityResult,
  7. GraphRelationshipResult,
  8. GraphSearchResult,
  9. GraphSearchResultType,
  10. GraphSearchSettings,
  11. HybridSearchSettings,
  12. IngestionMode,
  13. Message,
  14. MessageType,
  15. R2RException,
  16. R2RSerializable,
  17. SearchMode,
  18. SearchSettings,
  19. Token,
  20. User,
  21. select_search_filters,
  22. )
  23. from shared.abstractions.graph import (
  24. GraphCreationSettings,
  25. GraphEnrichmentSettings,
  26. )
  27. from shared.api.models import (
  28. AgentEvent,
  29. AgentResponse,
  30. Citation,
  31. CitationData,
  32. CitationEvent,
  33. Delta,
  34. DeltaPayload,
  35. FinalAnswerData,
  36. FinalAnswerEvent,
  37. MessageData,
  38. MessageDelta,
  39. MessageEvent,
  40. RAGResponse,
  41. SearchResultsData,
  42. SearchResultsEvent,
  43. SSEEventBase,
  44. ThinkingData,
  45. ThinkingEvent,
  46. ToolCallData,
  47. ToolCallEvent,
  48. ToolResultData,
  49. ToolResultEvent,
  50. UnknownEvent,
  51. )
  52. __all__ = [
  53. "AggregateSearchResult",
  54. "GenerationConfig",
  55. "HybridSearchSettings",
  56. "GraphCommunityResult",
  57. "GraphCreationSettings",
  58. "GraphEnrichmentSettings",
  59. "GraphEntityResult",
  60. "GraphRelationshipResult",
  61. "GraphSearchResult",
  62. "GraphSearchResultType",
  63. "GraphSearchSettings",
  64. "Message",
  65. "MessageType",
  66. "R2RException",
  67. "R2RSerializable",
  68. "Token",
  69. "ChunkSearchResult",
  70. "SearchSettings",
  71. "select_search_filters",
  72. "IngestionMode",
  73. "SearchMode",
  74. # "RAGResponse",
  75. "Citation",
  76. "RAGResponse",
  77. "AgentEvent",
  78. "AgentResponse",
  79. "SSEEventBase",
  80. "SearchResultsData",
  81. "SearchResultsEvent",
  82. "MessageData",
  83. "MessageDelta",
  84. "MessageEvent",
  85. "DeltaPayload",
  86. "Delta",
  87. "CitationData",
  88. "CitationEvent",
  89. "FinalAnswerData",
  90. "FinalAnswerEvent",
  91. "ToolCallData",
  92. "ToolCallEvent",
  93. "ToolResultData",
  94. "ToolResultEvent",
  95. "ThinkingEvent",
  96. "ThinkingData",
  97. "UnknownEvent",
  98. "User",
  99. ]