All files / src/platform/ai/schemas ai-gateway.schema.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
import { Document, Schema as MongooseSchema } from 'mongoose';
 
@Schema({ collection: 'ai_gateway_providers', timestamps: true })
export class AiGatewayProvider extends Document {
  @Prop({ required: true, unique: true })
  providerKey: string; // 'openai' | 'gemini' | 'claude' | 'deepseek' | 'ollama'
 
  @Prop({ required: true })
  name: string;
 
  @Prop({ default: true })
  isActive: boolean;
 
  @Prop({ type: Object, default: {} })
  config: Record<string, any>; // API endpoints, keys
}
export const AiGatewayProviderSchema =
  SchemaFactory.createForClass(AiGatewayProvider);
 
@Schema({ collection: 'ai_routing_policies', timestamps: true })
export class AiRoutingPolicy extends Document {
  @Prop({ required: true, unique: true })
  capability: string; // 'text_generation' | 'embeddings' | 'classification'
 
  @Prop({ required: true })
  primaryModelKey: string;
 
  @Prop({ type: [String], default: [] })
  fallbackModelKeys: string[];
 
  @Prop({ default: true })
  isActive: boolean;
}
export const AiRoutingPolicySchema =
  SchemaFactory.createForClass(AiRoutingPolicy);
 
@Schema({ collection: 'ai_credit_ledgers', timestamps: true })
export class AiCreditLedger extends Document {
  @Prop({ required: true, index: true })
  tenantId: string;
 
  @Prop({ required: true })
  amount: number; // credits balance
 
  @Prop({ required: true })
  transactionType: string; // 'credit' (purchase) or 'debit' (usage)
 
  @Prop({ required: true })
  description: string;
 
  createdAt: Date;
}
export const AiCreditLedgerSchema =
  SchemaFactory.createForClass(AiCreditLedger);
 
@Schema({ collection: 'ai_conversations', timestamps: true })
export class AiConversation extends Document {
  @Prop({ required: true, index: true })
  tenantId: string;
 
  @Prop({ required: true, index: true })
  userId: string;
 
  @Prop({ default: 'New Conversation' })
  title: string;
 
  @Prop({ default: null })
  systemPrompt: string;
 
  createdAt: Date;
  updatedAt: Date;
}
export const AiConversationSchema =
  SchemaFactory.createForClass(AiConversation);
 
@Schema({ collection: 'ai_messages', timestamps: true })
export class AiMessage extends Document {
  @Prop({ required: true, index: true })
  conversationId: string;
 
  @Prop({ required: true, enum: ['system', 'user', 'assistant', 'tool'] })
  role: string;
 
  @Prop({ required: true })
  content: string;
 
  @Prop({ type: Object, default: null })
  toolCalls: any;
 
  createdAt: Date;
}
export const AiMessageSchema = SchemaFactory.createForClass(AiMessage);
AiMessageSchema.index({ conversationId: 1, createdAt: 1 });