Coverage for src / lexigram / contracts / ai / __init__.py: 0%

30 statements  

« prev     ^ index     » next       coverage.py v7.13.5, created at 2026-08-19 05:41 +0800

1"""AI/ML protocols.""" 

2 

3from __future__ import annotations 

4 

5from lexigram.contracts.ai.agents import ( 

6 AgentError, 

7 AgentExecutionContext, 

8 AgentExecutorProtocol, 

9 AgentProtocol, 

10 AgentResponse, 

11 AgentStrategyProtocol, 

12 MemoryProtocol, 

13 SkillComposerProtocol, 

14 StrategyError, 

15 StrategyProtocol, 

16 ToolDefinition, 

17 ToolError, 

18 ToolProtocol, 

19 ToolRegistryProtocol, 

20 ToolResult, 

21) 

22from lexigram.contracts.ai.callbacks import ( 

23 CallbackHandlerProtocol, 

24 CallbackManagerProtocol, 

25) 

26from lexigram.contracts.ai.chunks import Chunk, ContextChunk 

27from lexigram.contracts.ai.evaluation import ( 

28 EvaluationDataset, 

29 EvaluationError, 

30 EvaluationHarnessProtocol, 

31 EvaluationResult, 

32 EvaluationSample, 

33 EvaluationScoreType, 

34 EvaluatorProtocol, 

35 RunReport, 

36) 

37from lexigram.contracts.ai.exceptions import ( 

38 AIError, 

39 AIMemoryError, 

40 ExtractionError, 

41 GuardError, 

42 LLMError, 

43 RAGError, 

44 RelayError, 

45 RelayErrorCode, 

46 RetrieverError, 

47 SkillError, 

48) 

49from lexigram.contracts.ai.governance import ( 

50 AIAuditEvent, 

51 AIAuditStoreProtocol, 

52 AIGovernanceProtocol, 

53 AuditEventType, 

54 CostTrackingProtocol, 

55 GovernanceDecision, 

56 ResourceExhaustedError, 

57 ResourceQuota, 

58 ResourceUnit, 

59 ResourceUsageResult, 

60 ResourceUsageSnapshot, 

61 ResourceWindowKind, 

62) 

63from lexigram.contracts.ai.guards import ( 

64 GuardPipelineProtocol, 

65 GuardResultProtocol, 

66 InputGuardProtocol, 

67 OutputGuardProtocol, 

68) 

69from lexigram.contracts.ai.index import ( 

70 Citation, 

71 IndexProtocol, 

72 QueryEngineError, 

73 QueryEngineProtocol, 

74 QueryEngineResponse, 

75) 

76from lexigram.contracts.ai.index import ( 

77 IndexError as IndexError_, 

78) 

79 

80IndexError = IndexError_ # noqa: A001 

81from lexigram.contracts.ai.llm import ( 

82 ChatMessage, 

83 ChatMessageProtocol, 

84 Completion, 

85 CompletionProtocol, 

86 CostEstimatorProtocol, 

87 EmbeddingClientProtocol, 

88 FunctionCall, 

89 LLMClientProtocol, 

90 PromptAssemblerProtocol, 

91 PromptRegistryProtocol, 

92 PromptTemplateProtocol, 

93 Role, 

94 SemanticCacheProtocol, 

95 StructuredExtractorProtocol, 

96 TokenBudget, 

97 TokenCounterProtocol, 

98 ToolCall, 

99) 

100from lexigram.contracts.ai.loaders import ( 

101 DocumentLoaderProtocol, 

102 LoaderError, 

103 LoaderRegistryProtocol, 

104) 

105from lexigram.contracts.ai.memory import ( 

106 ConsolidationResult, 

107 EpisodicMemoryProtocol, 

108 MemoryConsolidatorProtocol, 

109 MemoryEntry, 

110 MemoryQuery, 

111 MemorySearchResult, 

112 MemoryStoreProtocol, 

113 SemanticMemoryProtocol, 

114 WorkingMemoryProtocol, 

115) 

116from lexigram.contracts.ai.models import ( 

117 ModelRequest, 

118 ModelResponse, 

119) 

120from lexigram.contracts.ai.multimodal import ( 

121 ContentPart, 

122 ImageBase64Part, 

123 ImageUrlPart, 

124 MessageContent, 

125 TextPart, 

126) 

127from lexigram.contracts.ai.protocols import ( 

128 AIProviderProtocol, 

129 AISubsystemProtocol, 

130) 

131from lexigram.contracts.ai.providers import ( 

132 FallbackChainProtocol, 

133 ModelInfo, 

134 ModelSelectorProtocol, 

135 ProviderHealth, 

136 ProviderRegistryProtocol, 

137 SelectionStrategy, 

138) 

139from lexigram.contracts.ai.rag import ( 

140 PromptCompressorProtocol, 

141 RAGPipelineProtocol, 

142) 

143from lexigram.contracts.ai.relay import ( 

144 ClaudeContent, 

145 ClaudeMessage, 

146 ClaudeOptions, 

147 ClaudeRequest, 

148 ConversionQuality, 

149 GeminiContent, 

150 GeminiOptions, 

151 GeminiPart, 

152 GeminiRequest, 

153 JsonObject, 

154 JsonValue, 

155 MediaResolverProtocol, 

156 OpenAIChatMessage, 

157 OpenAIChatRequest, 

158 RelayConversionContext, 

159 RelayConverterProtocol, 

160 RelayConvertResult, 

161 RelayFormat, 

162 RelayLoss, 

163 RelayMapperProtocol, 

164 RelayOptions, 

165 RelayRegistryProtocol, 

166 RelayRequest, 

167 RelayRequestPayload, 

168 RelayResponse, 

169 RelayResponsePayload, 

170 RelayStreamOptions, 

171 RelayStreamSessionProtocol, 

172 RelayUsage, 

173 ResponsesItem, 

174 ResponsesRequest, 

175 ResponsesResponse, 

176 StreamDelta, 

177 StreamState, 

178) 

179from lexigram.contracts.ai.retrievers import ( 

180 NodePostprocessorProtocol, 

181 RetrievalQuery, 

182 RetrievedNode, 

183 RetrieverError, 

184 RetrieverProtocol, 

185) 

186from lexigram.contracts.ai.routing import ( 

187 InferenceLoggerProtocol, 

188 LLMRouterProtocol, 

189 QuotaBackendProtocol, 

190) 

191from lexigram.contracts.ai.session import ( 

192 ContextPrunerProtocol, 

193 SessionCheckpoint, 

194 SessionContextProtocol, 

195 SessionManagerProtocol, 

196 SessionState, 

197 SessionStatus, 

198 SessionStoreProtocol, 

199 SessionTurn, 

200) 

201from lexigram.contracts.ai.skills import ( 

202 SkillDefinition, 

203 SkillError, 

204 SkillExecutorProtocol, 

205 SkillParameter, 

206 SkillProtocol, 

207 SkillRegistryProtocol, 

208 SkillResult, 

209) 

210from lexigram.contracts.ai.thinking import ( 

211 ThinkingConfig, 

212 ThinkingResult, 

213) 

214from lexigram.contracts.ai.types import ( 

215 EmbeddingModel, 

216 ModelCapability, 

217 ModelProvider, 

218 VectorProvider, 

219) 

220from lexigram.contracts.ai.vector import ( 

221 ChunkerProtocol, 

222 Document, 

223 DocumentProtocol, 

224 DocumentVectorStoreProtocol, 

225 RAGSearchResult, 

226) 

227from lexigram.contracts.ai.workflow import ( 

228 AIWorkflowNodeProtocol, 

229 WorkflowProtocol, 

230) 

231from lexigram.contracts.observability.ai import ( 

232 AIHealthMonitorProtocol, 

233 AIMetricsProtocol, 

234 AITracerProtocol, 

235 ObservabilityProtocol, 

236) 

237 

238__all__ = [ 

239 "AIAuditEvent", 

240 "AIAuditStoreProtocol", 

241 "AIError", 

242 "AIGovernanceProtocol", 

243 "AIHealthMonitorProtocol", 

244 "AIMemoryError", 

245 "AIMetricsProtocol", 

246 "AIProviderProtocol", 

247 "AISubsystemProtocol", 

248 "AITracerProtocol", 

249 "AIWorkflowNodeProtocol", 

250 "AgentError", 

251 "AgentExecutionContext", 

252 "AgentExecutorProtocol", 

253 "AgentProtocol", 

254 "AgentResponse", 

255 "AgentStrategyProtocol", 

256 "AuditEventType", 

257 "CallbackHandlerProtocol", 

258 "CallbackManagerProtocol", 

259 "ChatMessage", 

260 "ChatMessageProtocol", 

261 "Chunk", 

262 "ChunkerProtocol", 

263 "Citation", 

264 "ClaudeContent", 

265 "ClaudeMessage", 

266 "ClaudeOptions", 

267 "ClaudeRequest", 

268 "Completion", 

269 "CompletionProtocol", 

270 "ConsolidationResult", 

271 "ContentPart", 

272 "ContextChunk", 

273 "ContextPrunerProtocol", 

274 "ConversionQuality", 

275 "CostEstimatorProtocol", 

276 "CostTrackingProtocol", 

277 "Document", 

278 "DocumentLoaderProtocol", 

279 "DocumentProtocol", 

280 "DocumentVectorStoreProtocol", 

281 "EmbeddingClientProtocol", 

282 "EmbeddingModel", 

283 "EpisodicMemoryProtocol", 

284 "EvaluationDataset", 

285 "EvaluationError", 

286 "EvaluationHarnessProtocol", 

287 "EvaluationResult", 

288 "EvaluationSample", 

289 "EvaluationScoreType", 

290 "EvaluatorProtocol", 

291 "ExtractionError", 

292 "FallbackChainProtocol", 

293 "FunctionCall", 

294 "GeminiContent", 

295 "GeminiOptions", 

296 "GeminiPart", 

297 "GeminiRequest", 

298 "GovernanceDecision", 

299 "GuardError", 

300 "GuardPipelineProtocol", 

301 "GuardResultProtocol", 

302 "ImageBase64Part", 

303 "ImageUrlPart", 

304 "IndexError", # noqa: A001 

305 "IndexProtocol", 

306 "InferenceLoggerProtocol", 

307 "InputGuardProtocol", 

308 "JsonObject", 

309 "JsonValue", 

310 "LLMClientProtocol", 

311 "LLMError", 

312 "LLMRouterProtocol", 

313 "LoaderError", 

314 "LoaderRegistryProtocol", 

315 "MediaResolverProtocol", 

316 "MemoryConsolidatorProtocol", 

317 "MemoryEntry", 

318 "MemoryProtocol", 

319 "MemoryQuery", 

320 "MemorySearchResult", 

321 "MemoryStoreProtocol", 

322 "MessageContent", 

323 "ModelCapability", 

324 "ModelInfo", 

325 "ModelProvider", 

326 "ModelRequest", 

327 "ModelResponse", 

328 "ModelSelectorProtocol", 

329 "NodePostprocessorProtocol", 

330 "ObservabilityProtocol", 

331 "OpenAIChatMessage", 

332 "OpenAIChatRequest", 

333 "OutputGuardProtocol", 

334 "PromptAssemblerProtocol", 

335 "PromptCompressorProtocol", 

336 "PromptRegistryProtocol", 

337 "PromptTemplateProtocol", 

338 "ProviderHealth", 

339 "ProviderRegistryProtocol", 

340 "QueryEngineError", 

341 "QueryEngineProtocol", 

342 "QueryEngineResponse", 

343 "QuotaBackendProtocol", 

344 "RAGError", 

345 "RAGPipelineProtocol", 

346 "RAGSearchResult", 

347 "RelayConversionContext", 

348 "RelayConvertResult", 

349 "RelayConverterProtocol", 

350 "RelayError", 

351 "RelayErrorCode", 

352 "RelayFormat", 

353 "RelayLoss", 

354 "RelayMapperProtocol", 

355 "RelayOptions", 

356 "RelayRegistryProtocol", 

357 "RelayRequest", 

358 "RelayRequestPayload", 

359 "RelayResponse", 

360 "RelayResponsePayload", 

361 "RelayStreamOptions", 

362 "RelayStreamSessionProtocol", 

363 "RelayUsage", 

364 "ResourceExhaustedError", 

365 "ResourceQuota", 

366 "ResourceUnit", 

367 "ResourceUsageResult", 

368 "ResourceUsageSnapshot", 

369 "ResourceWindowKind", 

370 "ResponsesItem", 

371 "ResponsesRequest", 

372 "ResponsesResponse", 

373 "RetrievalQuery", 

374 "RetrievedNode", 

375 "RetrieverError", 

376 "RetrieverProtocol", 

377 "Role", 

378 "RunReport", 

379 "SelectionStrategy", 

380 "SemanticCacheProtocol", 

381 "SemanticMemoryProtocol", 

382 "SessionCheckpoint", 

383 "SessionContextProtocol", 

384 "SessionManagerProtocol", 

385 "SessionState", 

386 "SessionStatus", 

387 "SessionStoreProtocol", 

388 "SessionTurn", 

389 "SkillComposerProtocol", 

390 "SkillDefinition", 

391 "SkillError", 

392 "SkillExecutorProtocol", 

393 "SkillParameter", 

394 "SkillProtocol", 

395 "SkillRegistryProtocol", 

396 "SkillResult", 

397 "StrategyError", 

398 "StrategyProtocol", 

399 "StreamDelta", 

400 "StreamState", 

401 "StructuredExtractorProtocol", 

402 "TextPart", 

403 "ThinkingConfig", 

404 "ThinkingResult", 

405 "TokenBudget", 

406 "TokenCounterProtocol", 

407 "ToolCall", 

408 "ToolDefinition", 

409 "ToolError", 

410 "ToolProtocol", 

411 "ToolRegistryProtocol", 

412 "ToolResult", 

413 "VectorProvider", 

414 "WorkflowProtocol", 

415 "WorkingMemoryProtocol", 

416]