Abdul Rahman
101 lines
2.7 KiB
TypeScript
101 lines
2.7 KiB
TypeScript
import {
|
|
Column,
|
|
CreateDateColumn,
|
|
DeleteDateColumn,
|
|
Entity,
|
|
Index,
|
|
JoinColumn,
|
|
ManyToOne,
|
|
OneToMany,
|
|
PrimaryGeneratedColumn,
|
|
UpdateDateColumn,
|
|
} from 'typeorm';
|
|
|
|
import { Relation } from 'src/engine/workspace-manager/workspace-sync-metadata/interfaces/relation.interface';
|
|
|
|
import { ModelId } from 'src/engine/core-modules/ai/constants/ai-models.const';
|
|
import { ApplicationEntity } from 'src/engine/core-modules/application/application.entity';
|
|
import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity';
|
|
import { ModelConfiguration } from 'src/engine/metadata-modules/agent/types/modelConfiguration';
|
|
|
|
import { AgentChatThreadEntity } from './agent-chat-thread.entity';
|
|
import { AgentHandoffEntity } from './agent-handoff.entity';
|
|
|
|
@Entity('agent')
|
|
@Index('IDX_AGENT_ID_DELETED_AT', ['id', 'deletedAt'])
|
|
@Index('IDX_AGENT_NAME_WORKSPACE_ID_UNIQUE', ['name', 'workspaceId'], {
|
|
unique: true,
|
|
where: '"deletedAt" IS NULL',
|
|
})
|
|
export class AgentEntity {
|
|
@PrimaryGeneratedColumn('uuid')
|
|
id: string;
|
|
|
|
@Column({ nullable: true, type: 'uuid' })
|
|
standardId?: string;
|
|
|
|
@Column({ nullable: false })
|
|
name: string;
|
|
|
|
@Column({ nullable: false })
|
|
label: string;
|
|
|
|
@Column({ nullable: true })
|
|
icon: string;
|
|
|
|
@Column({ nullable: true })
|
|
description: string;
|
|
|
|
@Column({ nullable: false, type: 'text' })
|
|
prompt: string;
|
|
|
|
@Column({ nullable: false, type: 'varchar', default: 'auto' })
|
|
modelId: ModelId;
|
|
|
|
@Column({ nullable: true, type: 'jsonb' })
|
|
responseFormat: object;
|
|
|
|
@Column({ nullable: false, type: 'uuid' })
|
|
workspaceId: string;
|
|
|
|
@Column({ nullable: true, type: 'uuid' })
|
|
applicationId: string | null;
|
|
|
|
@Column({ default: false })
|
|
isCustom: boolean;
|
|
|
|
@ManyToOne(() => Workspace, (workspace) => workspace.agents, {
|
|
onDelete: 'CASCADE',
|
|
})
|
|
@JoinColumn({ name: 'workspaceId' })
|
|
workspace: Relation<Workspace>;
|
|
|
|
@ManyToOne(() => ApplicationEntity, (application) => application.agents, {
|
|
onDelete: 'CASCADE',
|
|
nullable: true,
|
|
})
|
|
@JoinColumn({ name: 'applicationId' })
|
|
application: Relation<ApplicationEntity> | null;
|
|
|
|
@OneToMany(() => AgentChatThreadEntity, (chatThread) => chatThread.agent)
|
|
chatThreads: Relation<AgentChatThreadEntity[]>;
|
|
|
|
@OneToMany(() => AgentHandoffEntity, (handoff) => handoff.fromAgent)
|
|
outgoingHandoffs: Relation<AgentHandoffEntity[]>;
|
|
|
|
@OneToMany(() => AgentHandoffEntity, (handoff) => handoff.toAgent)
|
|
incomingHandoffs: Relation<AgentHandoffEntity[]>;
|
|
|
|
@CreateDateColumn({ type: 'timestamptz' })
|
|
createdAt: Date;
|
|
|
|
@UpdateDateColumn({ type: 'timestamptz' })
|
|
updatedAt: Date;
|
|
|
|
@DeleteDateColumn({ type: 'timestamptz' })
|
|
deletedAt?: Date;
|
|
|
|
@Column({ nullable: true, type: 'jsonb' })
|
|
modelConfiguration: ModelConfiguration;
|
|
}
|