refactor(store,tools): migrate IDs from UUID to bigserial int64
Some checks failed
CI / build-and-test (push) Failing after -31m12s
Some checks failed
CI / build-and-test (push) Failing after -31m12s
All internal entity lookups now use bigserial primary keys (int64) while GUIDs are retained for external/public identification. Updated store functions (TouchProject, UpdateThoughtMetadata, AddThoughtAttachment) to query by id instead of guid, added GetThoughtByID, changed semanticSearch and all tool helpers to use *int64 project IDs, and updated retry/backfill workers to use int64 thought IDs throughout.
This commit is contained in:
@@ -9,10 +9,11 @@ import (
|
||||
|
||||
type ModelPublicAgentGuardrails struct {
|
||||
bun.BaseModel `bun:"table:public.agent_guardrails,alias:agent_guardrails"`
|
||||
ID resolvespec_common.SqlUUID `bun:"id,type:uuid,pk,default:gen_random_uuid()," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
Content resolvespec_common.SqlString `bun:"content,type:text,notnull," json:"content"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||
Severity resolvespec_common.SqlString `bun:"severity,type:text,default:'medium',notnull," json:"severity"`
|
||||
Tags resolvespec_common.SqlString `bun:"tags,type:text,nullzero," json:"tags"`
|
||||
|
||||
@@ -9,10 +9,11 @@ import (
|
||||
|
||||
type ModelPublicAgentSkills struct {
|
||||
bun.BaseModel `bun:"table:public.agent_skills,alias:agent_skills"`
|
||||
ID resolvespec_common.SqlUUID `bun:"id,type:uuid,pk,default:gen_random_uuid()," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
Content resolvespec_common.SqlString `bun:"content,type:text,notnull," json:"content"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||
Tags resolvespec_common.SqlString `bun:"tags,type:text,nullzero," json:"tags"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||
|
||||
@@ -9,18 +9,19 @@ import (
|
||||
|
||||
type ModelPublicChatHistories struct {
|
||||
bun.BaseModel `bun:"table:public.chat_histories,alias:chat_histories"`
|
||||
ID resolvespec_common.SqlUUID `bun:"id,type:uuid,pk,default:gen_random_uuid()," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
AgentID resolvespec_common.SqlString `bun:"agent_id,type:text,nullzero," json:"agent_id"`
|
||||
Channel resolvespec_common.SqlString `bun:"channel,type:text,nullzero," json:"channel"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Messages resolvespec_common.SqlJSONB `bun:"messages,type:jsonb,default:'[',notnull," json:"messages"`
|
||||
Metadata resolvespec_common.SqlJSONB `bun:"metadata,type:jsonb,default:'{}',notnull," json:"metadata"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,nullzero," json:"project_id"`
|
||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||
SessionID resolvespec_common.SqlString `bun:"session_id,type:text,notnull," json:"session_id"`
|
||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,nullzero," json:"summary"`
|
||||
Title resolvespec_common.SqlString `bun:"title,type:text,nullzero," json:"title"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicChatHistories
|
||||
|
||||
@@ -15,9 +15,9 @@ type ModelPublicEmbeddings struct {
|
||||
Embedding resolvespec_common.SqlString `bun:"embedding,type:vector,notnull," json:"embedding"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Model resolvespec_common.SqlString `bun:"model,type:text,notnull,unique:uidx_embeddings_thought_id_model," json:"model"`
|
||||
ThoughtID resolvespec_common.SqlUUID `bun:"thought_id,type:uuid,notnull,unique:uidx_embeddings_thought_id_model," json:"thought_id"`
|
||||
ThoughtID int64 `bun:"thought_id,type:bigint,notnull,unique:uidx_embeddings_thought_id_model," json:"thought_id"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=guid" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicEmbeddings
|
||||
|
||||
@@ -9,31 +9,32 @@ import (
|
||||
|
||||
type ModelPublicLearnings struct {
|
||||
bun.BaseModel `bun:"table:public.learnings,alias:learnings"`
|
||||
ID resolvespec_common.SqlUUID `bun:"id,type:uuid,pk,default:gen_random_uuid()," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
ActionRequired bool `bun:"action_required,type:boolean,default:false,notnull," json:"action_required"`
|
||||
Area resolvespec_common.SqlString `bun:"area,type:text,default:'other',notnull," json:"area"`
|
||||
Category resolvespec_common.SqlString `bun:"category,type:text,default:'insight',notnull," json:"category"`
|
||||
Confidence resolvespec_common.SqlString `bun:"confidence,type:text,default:'hypothesis',notnull," json:"confidence"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
Details resolvespec_common.SqlString `bun:"details,type:text,default:'',notnull," json:"details"`
|
||||
DuplicateOfLearningID resolvespec_common.SqlUUID `bun:"duplicate_of_learning_id,type:uuid,nullzero," json:"duplicate_of_learning_id"`
|
||||
DuplicateOfLearningID resolvespec_common.SqlInt64 `bun:"duplicate_of_learning_id,type:bigint,nullzero," json:"duplicate_of_learning_id"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Priority resolvespec_common.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,nullzero," json:"project_id"`
|
||||
RelatedSkillID resolvespec_common.SqlUUID `bun:"related_skill_id,type:uuid,nullzero," json:"related_skill_id"`
|
||||
RelatedThoughtID resolvespec_common.SqlUUID `bun:"related_thought_id,type:uuid,nullzero," json:"related_thought_id"`
|
||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||
RelatedSkillID resolvespec_common.SqlInt64 `bun:"related_skill_id,type:bigint,nullzero," json:"related_skill_id"`
|
||||
RelatedThoughtID resolvespec_common.SqlInt64 `bun:"related_thought_id,type:bigint,nullzero," json:"related_thought_id"`
|
||||
ReviewedAt resolvespec_common.SqlTimeStamp `bun:"reviewed_at,type:timestamptz,nullzero," json:"reviewed_at"`
|
||||
ReviewedBy resolvespec_common.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
||||
SourceRef resolvespec_common.SqlString `bun:"source_ref,type:text,nullzero," json:"source_ref"`
|
||||
SourceType resolvespec_common.SqlString `bun:"source_type,type:text,nullzero," json:"source_type"`
|
||||
Status resolvespec_common.SqlString `bun:"status,type:text,default:'pending',notnull," json:"status"`
|
||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
||||
SupersedesLearningID resolvespec_common.SqlUUID `bun:"supersedes_learning_id,type:uuid,nullzero," json:"supersedes_learning_id"`
|
||||
SupersedesLearningID resolvespec_common.SqlInt64 `bun:"supersedes_learning_id,type:bigint,nullzero," json:"supersedes_learning_id"`
|
||||
Tags resolvespec_common.SqlString `bun:"tags,type:text,nullzero," json:"tags"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||
RelDuplicateOfLearningID *ModelPublicLearnings `bun:"rel:has-one,join:duplicate_of_learning_id=id" json:"relduplicateoflearningid,omitempty"` // Has one ModelPublicLearnings
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelRelatedSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:related_skill_id=id" json:"relrelatedskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||
RelRelatedThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:related_thought_id=guid" json:"relrelatedthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
RelRelatedThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:related_thought_id=id" json:"relrelatedthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
RelSupersedesLearningID *ModelPublicLearnings `bun:"rel:has-one,join:supersedes_learning_id=id" json:"relsupersedeslearningid,omitempty"` // Has one ModelPublicLearnings
|
||||
}
|
||||
|
||||
|
||||
@@ -9,10 +9,10 @@ import (
|
||||
|
||||
type ModelPublicPlanDependencies struct {
|
||||
bun.BaseModel `bun:"table:public.plan_dependencies,alias:plan_dependencies"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
DependsOnPlanID resolvespec_common.SqlUUID `bun:"depends_on_plan_id,type:uuid,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"depends_on_plan_id"`
|
||||
PlanID resolvespec_common.SqlUUID `bun:"plan_id,type:uuid,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"plan_id"`
|
||||
DependsOnPlanID int64 `bun:"depends_on_plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"depends_on_plan_id"`
|
||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"plan_id"`
|
||||
RelDependsOnPlanID *ModelPublicPlans `bun:"rel:has-one,join:depends_on_plan_id=id" json:"reldependsonplanid,omitempty"` // Has one ModelPublicPlans
|
||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||
}
|
||||
|
||||
@@ -9,10 +9,10 @@ import (
|
||||
|
||||
type ModelPublicPlanGuardrails struct {
|
||||
bun.BaseModel `bun:"table:public.plan_guardrails,alias:plan_guardrails"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
GuardrailID resolvespec_common.SqlUUID `bun:"guardrail_id,type:uuid,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"guardrail_id"`
|
||||
PlanID resolvespec_common.SqlUUID `bun:"plan_id,type:uuid,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"plan_id"`
|
||||
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"guardrail_id"`
|
||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"plan_id"`
|
||||
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||
}
|
||||
|
||||
@@ -9,10 +9,10 @@ import (
|
||||
|
||||
type ModelPublicPlanRelatedPlans struct {
|
||||
bun.BaseModel `bun:"table:public.plan_related_plans,alias:plan_related_plans"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
PlanAID resolvespec_common.SqlUUID `bun:"plan_a_id,type:uuid,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_a_id"`
|
||||
PlanBID resolvespec_common.SqlUUID `bun:"plan_b_id,type:uuid,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_b_id"`
|
||||
PlanAID int64 `bun:"plan_a_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_a_id"`
|
||||
PlanBID int64 `bun:"plan_b_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_b_id"`
|
||||
RelPlanAID *ModelPublicPlans `bun:"rel:has-one,join:plan_a_id=id" json:"relplanaid,omitempty"` // Has one ModelPublicPlans
|
||||
RelPlanBID *ModelPublicPlans `bun:"rel:has-one,join:plan_b_id=id" json:"relplanbid,omitempty"` // Has one ModelPublicPlans
|
||||
}
|
||||
|
||||
@@ -9,10 +9,10 @@ import (
|
||||
|
||||
type ModelPublicPlanSkills struct {
|
||||
bun.BaseModel `bun:"table:public.plan_skills,alias:plan_skills"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
PlanID resolvespec_common.SqlUUID `bun:"plan_id,type:uuid,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"plan_id"`
|
||||
SkillID resolvespec_common.SqlUUID `bun:"skill_id,type:uuid,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"skill_id"`
|
||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"plan_id"`
|
||||
SkillID int64 `bun:"skill_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"skill_id"`
|
||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||
}
|
||||
|
||||
@@ -9,22 +9,23 @@ import (
|
||||
|
||||
type ModelPublicPlans struct {
|
||||
bun.BaseModel `bun:"table:public.plans,alias:plans"`
|
||||
ID resolvespec_common.SqlUUID `bun:"id,type:uuid,pk,default:gen_random_uuid()," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CompletedAt resolvespec_common.SqlTimeStamp `bun:"completed_at,type:timestamptz,nullzero," json:"completed_at"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||
DueDate resolvespec_common.SqlTimeStamp `bun:"due_date,type:timestamptz,nullzero," json:"due_date"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
LastReviewedAt resolvespec_common.SqlTimeStamp `bun:"last_reviewed_at,type:timestamptz,nullzero," json:"last_reviewed_at"`
|
||||
Owner resolvespec_common.SqlString `bun:"owner,type:text,nullzero," json:"owner"`
|
||||
Priority resolvespec_common.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"` // low, medium, high, critical
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,nullzero," json:"project_id"`
|
||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||
ReviewedBy resolvespec_common.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
||||
Status resolvespec_common.SqlString `bun:"status,type:text,default:'draft',notnull," json:"status"` // draft, active, blocked, completed, cancelled, superseded
|
||||
SupersedesPlanID resolvespec_common.SqlUUID `bun:"supersedes_plan_id,type:uuid,nullzero," json:"supersedes_plan_id"`
|
||||
SupersedesPlanID resolvespec_common.SqlInt64 `bun:"supersedes_plan_id,type:bigint,nullzero," json:"supersedes_plan_id"`
|
||||
Tags resolvespec_common.SqlString `bun:"tags,type:text,nullzero," json:"tags"`
|
||||
Title resolvespec_common.SqlString `bun:"title,type:text,notnull," json:"title"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelSupersedesPlanID *ModelPublicPlans `bun:"rel:has-one,join:supersedes_plan_id=id" json:"relsupersedesplanid,omitempty"` // Has one ModelPublicPlans
|
||||
RelDependsOnPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=depends_on_plan_id" json:"reldependsonplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
||||
RelPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
||||
|
||||
@@ -9,12 +9,12 @@ import (
|
||||
|
||||
type ModelPublicProjectGuardrails struct {
|
||||
bun.BaseModel `bun:"table:public.project_guardrails,alias:project_guardrails"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
GuardrailID resolvespec_common.SqlUUID `bun:"guardrail_id,type:uuid,notnull," json:"guardrail_id"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,notnull," json:"project_id"`
|
||||
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull," json:"guardrail_id"`
|
||||
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
||||
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicProjectGuardrails
|
||||
|
||||
@@ -9,11 +9,11 @@ import (
|
||||
|
||||
type ModelPublicProjectSkills struct {
|
||||
bun.BaseModel `bun:"table:public.project_skills,alias:project_skills"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,notnull," json:"project_id"`
|
||||
SkillID resolvespec_common.SqlUUID `bun:"skill_id,type:uuid,notnull," json:"skill_id"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
||||
SkillID int64 `bun:"skill_id,type:bigint,notnull," json:"skill_id"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||
}
|
||||
|
||||
|
||||
@@ -16,13 +16,13 @@ type ModelPublicProjects struct {
|
||||
LastActiveAt resolvespec_common.SqlTimeStamp `bun:"last_active_at,type:timestamptz,default:now(),nullzero," json:"last_active_at"`
|
||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||
ThoughtCount resolvespec_common.SqlInt64 `bun:"thought_count,scanonly" json:"thought_count"`
|
||||
RelProjectIDPublicThoughts []*ModelPublicThoughts `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicthoughts,omitempty"` // Has many ModelPublicThoughts
|
||||
RelProjectIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||
RelProjectIDPublicChatHistories []*ModelPublicChatHistories `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicchathistories,omitempty"` // Has many ModelPublicChatHistories
|
||||
RelProjectIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||
RelProjectIDPublicPlans []*ModelPublicPlans `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicplans,omitempty"` // Has many ModelPublicPlans
|
||||
RelProjectIDPublicProjectSkills []*ModelPublicProjectSkills `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicprojectskills,omitempty"` // Has many ModelPublicProjectSkills
|
||||
RelProjectIDPublicProjectGuardrails []*ModelPublicProjectGuardrails `bun:"rel:has-many,join:guid=project_id" json:"relprojectidpublicprojectguardrails,omitempty"` // Has many ModelPublicProjectGuardrails
|
||||
RelProjectIDPublicThoughts []*ModelPublicThoughts `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicthoughts,omitempty"` // Has many ModelPublicThoughts
|
||||
RelProjectIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||
RelProjectIDPublicChatHistories []*ModelPublicChatHistories `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicchathistories,omitempty"` // Has many ModelPublicChatHistories
|
||||
RelProjectIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=project_id" json:"relprojectidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||
RelProjectIDPublicPlans []*ModelPublicPlans `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicplans,omitempty"` // Has many ModelPublicPlans
|
||||
RelProjectIDPublicProjectSkills []*ModelPublicProjectSkills `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicprojectskills,omitempty"` // Has many ModelPublicProjectSkills
|
||||
RelProjectIDPublicProjectGuardrails []*ModelPublicProjectGuardrails `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicprojectguardrails,omitempty"` // Has many ModelPublicProjectGuardrails
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicProjects
|
||||
|
||||
@@ -17,13 +17,13 @@ type ModelPublicStoredFiles struct {
|
||||
Kind resolvespec_common.SqlString `bun:"kind,type:text,default:'file',notnull," json:"kind"`
|
||||
MediaType resolvespec_common.SqlString `bun:"media_type,type:text,notnull," json:"media_type"`
|
||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,nullzero," json:"project_id"`
|
||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||
Sha256 resolvespec_common.SqlString `bun:"sha256,type:text,notnull," json:"sha256"`
|
||||
SizeBytes int64 `bun:"size_bytes,type:bigint,notnull," json:"size_bytes"`
|
||||
ThoughtID resolvespec_common.SqlUUID `bun:"thought_id,type:uuid,nullzero," json:"thought_id"`
|
||||
ThoughtID resolvespec_common.SqlInt64 `bun:"thought_id,type:bigint,nullzero," json:"thought_id"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=guid" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicStoredFiles
|
||||
|
||||
@@ -9,7 +9,7 @@ import (
|
||||
|
||||
type ModelPublicThoughtLinks struct {
|
||||
bun.BaseModel `bun:"table:public.thought_links,alias:thought_links"`
|
||||
ID resolvespec_common.SqlInt32 `bun:"id,type:serial,pk,autoincrement," json:"id"`
|
||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||
FromID int64 `bun:"from_id,type:bigint,notnull," json:"from_id"`
|
||||
Relation resolvespec_common.SqlString `bun:"relation,type:text,notnull," json:"relation"`
|
||||
|
||||
@@ -15,14 +15,14 @@ type ModelPublicThoughts struct {
|
||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||
Metadata resolvespec_common.SqlJSONB `bun:"metadata,type:jsonb,default:'{}::jsonb',nullzero," json:"metadata"`
|
||||
ProjectID resolvespec_common.SqlUUID `bun:"project_id,type:uuid,nullzero," json:"project_id"`
|
||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=guid" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||
RelFromIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=from_id" json:"relfromidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
||||
RelToIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=to_id" json:"reltoidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
||||
RelThoughtIDPublicEmbeddings []*ModelPublicEmbeddings `bun:"rel:has-many,join:guid=thought_id" json:"relthoughtidpublicembeddings,omitempty"` // Has many ModelPublicEmbeddings
|
||||
RelThoughtIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:guid=thought_id" json:"relthoughtidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||
RelRelatedThoughtIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:guid=related_thought_id" json:"relrelatedthoughtidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||
RelThoughtIDPublicEmbeddings []*ModelPublicEmbeddings `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicembeddings,omitempty"` // Has many ModelPublicEmbeddings
|
||||
RelThoughtIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||
RelRelatedThoughtIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=related_thought_id" json:"relrelatedthoughtidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||
}
|
||||
|
||||
// TableName returns the table name for ModelPublicThoughts
|
||||
|
||||
@@ -7,7 +7,6 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/jackc/pgx/v5"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/generatedmodels"
|
||||
@@ -28,7 +27,7 @@ func (db *DB) SaveChatHistory(ctx context.Context, h ext.ChatHistory) (ext.ChatH
|
||||
insert into chat_histories
|
||||
(session_id, title, channel, agent_id, project_id, messages, summary, metadata)
|
||||
values ($1, $2, $3, $4, $5, $6, $7, $8)
|
||||
returning id, created_at, updated_at
|
||||
returning id, guid, created_at, updated_at
|
||||
`,
|
||||
h.SessionID, nullStr(h.Title), nullStr(h.Channel), nullStr(h.AgentID),
|
||||
h.ProjectID, messages, nullStr(h.Summary), meta,
|
||||
@@ -36,16 +35,17 @@ func (db *DB) SaveChatHistory(ctx context.Context, h ext.ChatHistory) (ext.ChatH
|
||||
|
||||
created := h
|
||||
var model generatedmodels.ModelPublicChatHistories
|
||||
if err := row.Scan(&model.ID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return ext.ChatHistory{}, fmt.Errorf("insert chat history: %w", err)
|
||||
}
|
||||
created.ID = model.ID.UUID()
|
||||
created.ID = model.ID.Int64()
|
||||
created.GUID = model.GUID.UUID()
|
||||
created.CreatedAt = model.CreatedAt.Time()
|
||||
created.UpdatedAt = model.UpdatedAt.Time()
|
||||
return created, nil
|
||||
}
|
||||
|
||||
func (db *DB) GetChatHistory(ctx context.Context, id uuid.UUID) (ext.ChatHistory, bool, error) {
|
||||
func (db *DB) GetChatHistory(ctx context.Context, id int64) (ext.ChatHistory, bool, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select id, session_id, title, channel, agent_id, project_id,
|
||||
messages, summary, metadata, created_at, updated_at
|
||||
@@ -81,7 +81,7 @@ func (db *DB) GetChatHistoryBySessionID(ctx context.Context, sessionID string) (
|
||||
}
|
||||
|
||||
type ListChatHistoriesFilter struct {
|
||||
ProjectID *uuid.UUID
|
||||
ProjectID *int64
|
||||
Channel string
|
||||
AgentID string
|
||||
SessionID string
|
||||
@@ -149,7 +149,7 @@ func (db *DB) ListChatHistories(ctx context.Context, f ListChatHistoriesFilter)
|
||||
return result, rows.Err()
|
||||
}
|
||||
|
||||
func (db *DB) DeleteChatHistory(ctx context.Context, id uuid.UUID) (bool, error) {
|
||||
func (db *DB) DeleteChatHistory(ctx context.Context, id int64) (bool, error) {
|
||||
tag, err := db.pool.Exec(ctx, `delete from chat_histories where id = $1`, id)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("delete chat history: %w", err)
|
||||
@@ -171,7 +171,8 @@ func scanChatHistory(row rowScanner) (ext.ChatHistory, error) {
|
||||
}
|
||||
|
||||
h := ext.ChatHistory{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
GUID: model.GUID.UUID(),
|
||||
SessionID: model.SessionID.String(),
|
||||
Title: model.Title.String(),
|
||||
Channel: model.Channel.String(),
|
||||
@@ -181,7 +182,7 @@ func scanChatHistory(row rowScanner) (ext.ChatHistory, error) {
|
||||
UpdatedAt: model.UpdatedAt.Time(),
|
||||
}
|
||||
if model.ProjectID.Valid {
|
||||
id := model.ProjectID.UUID()
|
||||
id := model.ProjectID.Int64()
|
||||
h.ProjectID = &id
|
||||
}
|
||||
|
||||
|
||||
@@ -17,11 +17,12 @@ func (db *DB) InsertStoredFile(ctx context.Context, file thoughttypes.StoredFile
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
insert into stored_files (thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, content)
|
||||
values ($1, $2, $3, $4, $5, $6, $7, $8, $9)
|
||||
returning guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
||||
returning id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
||||
`, file.ThoughtID, file.ProjectID, file.Name, file.MediaType, file.Kind, file.Encoding, file.SizeBytes, file.SHA256, file.Content)
|
||||
|
||||
var model generatedmodels.ModelPublicStoredFiles
|
||||
if err := row.Scan(
|
||||
&model.ID,
|
||||
&model.GUID,
|
||||
&model.ThoughtID,
|
||||
&model.ProjectID,
|
||||
@@ -42,13 +43,14 @@ func (db *DB) InsertStoredFile(ctx context.Context, file thoughttypes.StoredFile
|
||||
|
||||
func (db *DB) GetStoredFile(ctx context.Context, id uuid.UUID) (thoughttypes.StoredFile, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, content, created_at, updated_at
|
||||
select id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, content, created_at, updated_at
|
||||
from stored_files
|
||||
where guid = $1
|
||||
`, id)
|
||||
|
||||
var model generatedmodels.ModelPublicStoredFiles
|
||||
if err := row.Scan(
|
||||
&model.ID,
|
||||
&model.GUID,
|
||||
&model.ThoughtID,
|
||||
&model.ProjectID,
|
||||
@@ -89,7 +91,7 @@ func (db *DB) ListStoredFiles(ctx context.Context, filter thoughttypes.StoredFil
|
||||
}
|
||||
|
||||
query := `
|
||||
select guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
||||
select id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
||||
from stored_files
|
||||
`
|
||||
if len(conditions) > 0 {
|
||||
@@ -109,6 +111,7 @@ func (db *DB) ListStoredFiles(ctx context.Context, filter thoughttypes.StoredFil
|
||||
for rows.Next() {
|
||||
var model generatedmodels.ModelPublicStoredFiles
|
||||
if err := rows.Scan(
|
||||
&model.ID,
|
||||
&model.GUID,
|
||||
&model.ThoughtID,
|
||||
&model.ProjectID,
|
||||
@@ -132,7 +135,7 @@ func (db *DB) ListStoredFiles(ctx context.Context, filter thoughttypes.StoredFil
|
||||
return files, nil
|
||||
}
|
||||
|
||||
func (db *DB) AddThoughtAttachment(ctx context.Context, thoughtID uuid.UUID, attachment thoughttypes.ThoughtAttachment) error {
|
||||
func (db *DB) AddThoughtAttachment(ctx context.Context, thoughtID int64, attachment thoughttypes.ThoughtAttachment) error {
|
||||
tx, err := db.pool.Begin(ctx)
|
||||
if err != nil {
|
||||
return fmt.Errorf("begin transaction: %w", err)
|
||||
@@ -142,7 +145,7 @@ func (db *DB) AddThoughtAttachment(ctx context.Context, thoughtID uuid.UUID, att
|
||||
}()
|
||||
|
||||
var metadataBytes []byte
|
||||
if err := tx.QueryRow(ctx, `select metadata from thoughts where guid = $1 for update`, thoughtID).Scan(&metadataBytes); err != nil {
|
||||
if err := tx.QueryRow(ctx, `select metadata from thoughts where id = $1 for update`, thoughtID).Scan(&metadataBytes); err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return err
|
||||
}
|
||||
@@ -177,7 +180,7 @@ func (db *DB) AddThoughtAttachment(ctx context.Context, thoughtID uuid.UUID, att
|
||||
update thoughts
|
||||
set metadata = $2::jsonb,
|
||||
updated_at = now()
|
||||
where guid = $1
|
||||
where id = $1
|
||||
`, thoughtID, updatedMetadata)
|
||||
if err != nil {
|
||||
return fmt.Errorf("update thought attachments: %w", err)
|
||||
|
||||
@@ -6,7 +6,6 @@ import (
|
||||
"strings"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/generatedmodels"
|
||||
"github.com/google/uuid"
|
||||
"github.com/jackc/pgx/v5"
|
||||
|
||||
thoughttypes "git.warky.dev/wdevs/amcs/internal/types"
|
||||
@@ -25,7 +24,7 @@ func (db *DB) CreateLearning(ctx context.Context, learning thoughttypes.Learning
|
||||
$13, $14, $15, $16,
|
||||
$17, $18
|
||||
)
|
||||
returning id, created_at, updated_at
|
||||
returning id, guid, created_at, updated_at
|
||||
`,
|
||||
strings.TrimSpace(learning.Summary),
|
||||
strings.TrimSpace(learning.Details),
|
||||
@@ -49,16 +48,17 @@ func (db *DB) CreateLearning(ctx context.Context, learning thoughttypes.Learning
|
||||
|
||||
created := learning
|
||||
var model generatedmodels.ModelPublicLearnings
|
||||
if err := row.Scan(&model.ID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return thoughttypes.Learning{}, fmt.Errorf("create learning: %w", err)
|
||||
}
|
||||
created.ID = model.ID.UUID()
|
||||
created.ID = model.ID.Int64()
|
||||
created.GUID = model.GUID.UUID()
|
||||
created.CreatedAt = model.CreatedAt.Time()
|
||||
created.UpdatedAt = model.UpdatedAt.Time()
|
||||
return created, nil
|
||||
}
|
||||
|
||||
func (db *DB) GetLearning(ctx context.Context, id uuid.UUID) (thoughttypes.Learning, error) {
|
||||
func (db *DB) GetLearning(ctx context.Context, id int64) (thoughttypes.Learning, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select id, summary, details, category, area, status, priority, confidence,
|
||||
action_required, source_type, source_ref, project_id, related_thought_id,
|
||||
@@ -71,7 +71,7 @@ func (db *DB) GetLearning(ctx context.Context, id uuid.UUID) (thoughttypes.Learn
|
||||
learning, err := scanLearning(row)
|
||||
if err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return thoughttypes.Learning{}, fmt.Errorf("learning not found: %s", id)
|
||||
return thoughttypes.Learning{}, fmt.Errorf("learning not found: %d", id)
|
||||
}
|
||||
return thoughttypes.Learning{}, fmt.Errorf("get learning: %w", err)
|
||||
}
|
||||
|
||||
@@ -25,12 +25,12 @@ func (db *DB) InsertLink(ctx context.Context, link thoughttypes.ThoughtLink) err
|
||||
|
||||
func (db *DB) LinkedThoughts(ctx context.Context, thoughtID uuid.UUID) ([]thoughttypes.LinkedThought, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at, l.relation, 'outgoing' as direction, l.created_at
|
||||
select t.id, t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at, l.relation, 'outgoing' as direction, l.created_at
|
||||
from thought_links l
|
||||
join thoughts t on t.id = l.to_id
|
||||
where l.from_id = (select id from thoughts where guid = $1)
|
||||
union all
|
||||
select t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at, l.relation, 'incoming' as direction, l.created_at
|
||||
select t.id, t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at, l.relation, 'incoming' as direction, l.created_at
|
||||
from thought_links l
|
||||
join thoughts t on t.id = l.from_id
|
||||
where l.to_id = (select id from thoughts where guid = $1)
|
||||
@@ -46,6 +46,7 @@ func (db *DB) LinkedThoughts(ctx context.Context, thoughtID uuid.UUID) ([]though
|
||||
var linked thoughttypes.LinkedThought
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := rows.Scan(
|
||||
&model.ID,
|
||||
&model.GUID,
|
||||
&model.Content,
|
||||
&model.Metadata,
|
||||
|
||||
@@ -5,8 +5,6 @@ import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/generatedmodels"
|
||||
ext "git.warky.dev/wdevs/amcs/internal/types"
|
||||
)
|
||||
@@ -14,6 +12,7 @@ import (
|
||||
func projectFromModel(m generatedmodels.ModelPublicProjects) ext.Project {
|
||||
return ext.Project{
|
||||
ID: m.GUID.UUID(),
|
||||
NumericID: m.ID.Int64(),
|
||||
Name: m.Name.String(),
|
||||
Description: m.Description.String(),
|
||||
CreatedAt: m.CreatedAt.Time(),
|
||||
@@ -29,9 +28,9 @@ func thoughtFromModel(m generatedmodels.ModelPublicThoughts) (ext.Thought, error
|
||||
}
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if m.ProjectID.Valid {
|
||||
id := m.ProjectID.UUID()
|
||||
id := m.ProjectID.Int64()
|
||||
projectID = &id
|
||||
}
|
||||
|
||||
@@ -42,7 +41,8 @@ func thoughtFromModel(m generatedmodels.ModelPublicThoughts) (ext.Thought, error
|
||||
}
|
||||
|
||||
return ext.Thought{
|
||||
ID: m.GUID.UUID(),
|
||||
ID: m.ID.Int64(),
|
||||
GUID: m.GUID.UUID(),
|
||||
Content: m.Content.String(),
|
||||
Metadata: metadata,
|
||||
ProjectID: projectID,
|
||||
@@ -53,20 +53,21 @@ func thoughtFromModel(m generatedmodels.ModelPublicThoughts) (ext.Thought, error
|
||||
}
|
||||
|
||||
func storedFileFromModel(m generatedmodels.ModelPublicStoredFiles) ext.StoredFile {
|
||||
var thoughtID *uuid.UUID
|
||||
var thoughtID *int64
|
||||
if m.ThoughtID.Valid {
|
||||
id := m.ThoughtID.UUID()
|
||||
id := m.ThoughtID.Int64()
|
||||
thoughtID = &id
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if m.ProjectID.Valid {
|
||||
id := m.ProjectID.UUID()
|
||||
id := m.ProjectID.Int64()
|
||||
projectID = &id
|
||||
}
|
||||
|
||||
return ext.StoredFile{
|
||||
ID: m.GUID.UUID(),
|
||||
ID: m.ID.Int64(),
|
||||
GUID: m.GUID.UUID(),
|
||||
ThoughtID: thoughtID,
|
||||
ProjectID: projectID,
|
||||
Name: m.Name.String(),
|
||||
@@ -419,9 +420,9 @@ func storedFileFromModel(m generatedmodels.ModelPublicStoredFiles) ext.StoredFil
|
||||
// }
|
||||
|
||||
func planFromModel(m generatedmodels.ModelPublicPlans, tags []string) ext.Plan {
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if m.ProjectID.Valid {
|
||||
id := m.ProjectID.UUID()
|
||||
id := m.ProjectID.Int64()
|
||||
projectID = &id
|
||||
}
|
||||
|
||||
@@ -443,14 +444,15 @@ func planFromModel(m generatedmodels.ModelPublicPlans, tags []string) ext.Plan {
|
||||
lastReviewedAt = &t
|
||||
}
|
||||
|
||||
var supersedesPlanID *uuid.UUID
|
||||
var supersedesPlanID *int64
|
||||
if m.SupersedesPlanID.Valid {
|
||||
id := m.SupersedesPlanID.UUID()
|
||||
id := m.SupersedesPlanID.Int64()
|
||||
supersedesPlanID = &id
|
||||
}
|
||||
|
||||
return ext.Plan{
|
||||
ID: m.ID.UUID(),
|
||||
ID: m.ID.Int64(),
|
||||
GUID: m.GUID.UUID(),
|
||||
Title: m.Title.String(),
|
||||
Description: m.Description.String(),
|
||||
Status: ext.PlanStatus(m.Status.String()),
|
||||
@@ -469,33 +471,33 @@ func planFromModel(m generatedmodels.ModelPublicPlans, tags []string) ext.Plan {
|
||||
}
|
||||
|
||||
func learningFromModel(m generatedmodels.ModelPublicLearnings, tags []string) ext.Learning {
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if m.ProjectID.Valid {
|
||||
id := m.ProjectID.UUID()
|
||||
id := m.ProjectID.Int64()
|
||||
projectID = &id
|
||||
}
|
||||
|
||||
var relatedThoughtID *uuid.UUID
|
||||
var relatedThoughtID *int64
|
||||
if m.RelatedThoughtID.Valid {
|
||||
id := m.RelatedThoughtID.UUID()
|
||||
id := m.RelatedThoughtID.Int64()
|
||||
relatedThoughtID = &id
|
||||
}
|
||||
|
||||
var relatedSkillID *uuid.UUID
|
||||
var relatedSkillID *int64
|
||||
if m.RelatedSkillID.Valid {
|
||||
id := m.RelatedSkillID.UUID()
|
||||
id := m.RelatedSkillID.Int64()
|
||||
relatedSkillID = &id
|
||||
}
|
||||
|
||||
var duplicateOfLearningID *uuid.UUID
|
||||
var duplicateOfLearningID *int64
|
||||
if m.DuplicateOfLearningID.Valid {
|
||||
id := m.DuplicateOfLearningID.UUID()
|
||||
id := m.DuplicateOfLearningID.Int64()
|
||||
duplicateOfLearningID = &id
|
||||
}
|
||||
|
||||
var supersedesLearningID *uuid.UUID
|
||||
var supersedesLearningID *int64
|
||||
if m.SupersedesLearningID.Valid {
|
||||
id := m.SupersedesLearningID.UUID()
|
||||
id := m.SupersedesLearningID.Int64()
|
||||
supersedesLearningID = &id
|
||||
}
|
||||
|
||||
@@ -512,7 +514,8 @@ func learningFromModel(m generatedmodels.ModelPublicLearnings, tags []string) ex
|
||||
}
|
||||
|
||||
return ext.Learning{
|
||||
ID: m.ID.UUID(),
|
||||
ID: m.ID.Int64(),
|
||||
GUID: m.GUID.UUID(),
|
||||
Summary: m.Summary.String(),
|
||||
Details: m.Details.String(),
|
||||
Category: m.Category.String(),
|
||||
|
||||
@@ -4,7 +4,7 @@ import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
"github.com/google/uuid"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/generatedmodels"
|
||||
@@ -12,7 +12,7 @@ import (
|
||||
)
|
||||
|
||||
const planColumns = `
|
||||
id, title, description, status, priority, project_id, owner, due_date,
|
||||
id, guid, title, description, status, priority, project_id, owner, due_date,
|
||||
completed_at, reviewed_by, last_reviewed_at, supersedes_plan_id, tags::text[], created_at, updated_at`
|
||||
|
||||
func (db *DB) CreatePlan(ctx context.Context, plan ext.Plan) (ext.Plan, error) {
|
||||
@@ -37,19 +37,19 @@ func (db *DB) CreatePlan(ctx context.Context, plan ext.Plan) (ext.Plan, error) {
|
||||
return scanPlan(row)
|
||||
}
|
||||
|
||||
func (db *DB) GetPlan(ctx context.Context, id uuid.UUID) (ext.Plan, error) {
|
||||
func (db *DB) GetPlan(ctx context.Context, id int64) (ext.Plan, error) {
|
||||
row := db.pool.QueryRow(ctx, `select`+planColumns+` from plans where id = $1`, id)
|
||||
plan, err := scanPlan(row)
|
||||
if err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return ext.Plan{}, fmt.Errorf("plan not found: %s", id)
|
||||
return ext.Plan{}, fmt.Errorf("plan not found: %d", id)
|
||||
}
|
||||
return ext.Plan{}, fmt.Errorf("get plan: %w", err)
|
||||
}
|
||||
return plan, nil
|
||||
}
|
||||
|
||||
func (db *DB) GetPlanDetail(ctx context.Context, id uuid.UUID) (ext.PlanDetail, error) {
|
||||
func (db *DB) GetPlanDetail(ctx context.Context, id int64) (ext.PlanDetail, error) {
|
||||
plan, err := db.GetPlan(ctx, id)
|
||||
if err != nil {
|
||||
return ext.PlanDetail{}, err
|
||||
@@ -105,7 +105,7 @@ func (db *DB) GetPlanDetail(ctx context.Context, id uuid.UUID) (ext.PlanDetail,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (db *DB) UpdatePlan(ctx context.Context, id uuid.UUID, u ext.PlanUpdate) (ext.Plan, error) {
|
||||
func (db *DB) UpdatePlan(ctx context.Context, id int64, u ext.PlanUpdate) (ext.Plan, error) {
|
||||
sets := []string{"updated_at = now()"}
|
||||
args := []any{}
|
||||
|
||||
@@ -169,14 +169,14 @@ func (db *DB) UpdatePlan(ctx context.Context, id uuid.UUID, u ext.PlanUpdate) (e
|
||||
plan, err := scanPlan(row)
|
||||
if err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return ext.Plan{}, fmt.Errorf("plan not found: %s", id)
|
||||
return ext.Plan{}, fmt.Errorf("plan not found: %d", id)
|
||||
}
|
||||
return ext.Plan{}, fmt.Errorf("update plan: %w", err)
|
||||
}
|
||||
return plan, nil
|
||||
}
|
||||
|
||||
func (db *DB) DeletePlan(ctx context.Context, id uuid.UUID) error {
|
||||
func (db *DB) DeletePlan(ctx context.Context, id int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `delete from plans where id = $1`, id)
|
||||
if err != nil {
|
||||
return fmt.Errorf("delete plan: %w", err)
|
||||
@@ -232,7 +232,7 @@ func (db *DB) ListPlans(ctx context.Context, filter ext.PlanFilter) ([]ext.Plan,
|
||||
|
||||
// Dependencies
|
||||
|
||||
func (db *DB) AddPlanDependency(ctx context.Context, planID, dependsOnPlanID uuid.UUID) error {
|
||||
func (db *DB) AddPlanDependency(ctx context.Context, planID, dependsOnPlanID int64) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into plan_dependencies (plan_id, depends_on_plan_id)
|
||||
values ($1, $2)
|
||||
@@ -244,7 +244,7 @@ func (db *DB) AddPlanDependency(ctx context.Context, planID, dependsOnPlanID uui
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemovePlanDependency(ctx context.Context, planID, dependsOnPlanID uuid.UUID) error {
|
||||
func (db *DB) RemovePlanDependency(ctx context.Context, planID, dependsOnPlanID int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from plan_dependencies where plan_id = $1 and depends_on_plan_id = $2
|
||||
`, planID, dependsOnPlanID)
|
||||
@@ -259,7 +259,7 @@ func (db *DB) RemovePlanDependency(ctx context.Context, planID, dependsOnPlanID
|
||||
|
||||
// Related Plans
|
||||
|
||||
func (db *DB) AddRelatedPlan(ctx context.Context, planAID, planBID uuid.UUID) error {
|
||||
func (db *DB) AddRelatedPlan(ctx context.Context, planAID, planBID int64) error {
|
||||
a, b := canonicalPlanPair(planAID, planBID)
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into plan_related_plans (plan_a_id, plan_b_id)
|
||||
@@ -272,7 +272,7 @@ func (db *DB) AddRelatedPlan(ctx context.Context, planAID, planBID uuid.UUID) er
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemoveRelatedPlan(ctx context.Context, planAID, planBID uuid.UUID) error {
|
||||
func (db *DB) RemoveRelatedPlan(ctx context.Context, planAID, planBID int64) error {
|
||||
a, b := canonicalPlanPair(planAID, planBID)
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from plan_related_plans where plan_a_id = $1 and plan_b_id = $2
|
||||
@@ -288,7 +288,7 @@ func (db *DB) RemoveRelatedPlan(ctx context.Context, planAID, planBID uuid.UUID)
|
||||
|
||||
// Plan Skills
|
||||
|
||||
func (db *DB) AddPlanSkill(ctx context.Context, planID, skillID uuid.UUID) error {
|
||||
func (db *DB) AddPlanSkill(ctx context.Context, planID, skillID int64) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into plan_skills (plan_id, skill_id) values ($1, $2) on conflict do nothing
|
||||
`, planID, skillID)
|
||||
@@ -298,7 +298,7 @@ func (db *DB) AddPlanSkill(ctx context.Context, planID, skillID uuid.UUID) error
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemovePlanSkill(ctx context.Context, planID, skillID uuid.UUID) error {
|
||||
func (db *DB) RemovePlanSkill(ctx context.Context, planID, skillID int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from plan_skills where plan_id = $1 and skill_id = $2
|
||||
`, planID, skillID)
|
||||
@@ -311,7 +311,7 @@ func (db *DB) RemovePlanSkill(ctx context.Context, planID, skillID uuid.UUID) er
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) ListPlanSkills(ctx context.Context, planID uuid.UUID) ([]ext.AgentSkill, error) {
|
||||
func (db *DB) ListPlanSkills(ctx context.Context, planID int64) ([]ext.AgentSkill, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select s.id, s.name, s.description, s.content, s.tags::text[], s.created_at, s.updated_at
|
||||
from agent_skills s
|
||||
@@ -332,7 +332,7 @@ func (db *DB) ListPlanSkills(ctx context.Context, planID uuid.UUID) ([]ext.Agent
|
||||
return nil, fmt.Errorf("scan plan skill: %w", err)
|
||||
}
|
||||
s := ext.AgentSkill{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -350,7 +350,7 @@ func (db *DB) ListPlanSkills(ctx context.Context, planID uuid.UUID) ([]ext.Agent
|
||||
|
||||
// Plan Guardrails
|
||||
|
||||
func (db *DB) AddPlanGuardrail(ctx context.Context, planID, guardrailID uuid.UUID) error {
|
||||
func (db *DB) AddPlanGuardrail(ctx context.Context, planID, guardrailID int64) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into plan_guardrails (plan_id, guardrail_id) values ($1, $2) on conflict do nothing
|
||||
`, planID, guardrailID)
|
||||
@@ -360,7 +360,7 @@ func (db *DB) AddPlanGuardrail(ctx context.Context, planID, guardrailID uuid.UUI
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemovePlanGuardrail(ctx context.Context, planID, guardrailID uuid.UUID) error {
|
||||
func (db *DB) RemovePlanGuardrail(ctx context.Context, planID, guardrailID int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from plan_guardrails where plan_id = $1 and guardrail_id = $2
|
||||
`, planID, guardrailID)
|
||||
@@ -373,7 +373,7 @@ func (db *DB) RemovePlanGuardrail(ctx context.Context, planID, guardrailID uuid.
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) ListPlanGuardrails(ctx context.Context, planID uuid.UUID) ([]ext.AgentGuardrail, error) {
|
||||
func (db *DB) ListPlanGuardrails(ctx context.Context, planID int64) ([]ext.AgentGuardrail, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select g.id, g.name, g.description, g.content, g.severity, g.tags::text[], g.created_at, g.updated_at
|
||||
from agent_guardrails g
|
||||
@@ -394,7 +394,7 @@ func (db *DB) ListPlanGuardrails(ctx context.Context, planID uuid.UUID) ([]ext.A
|
||||
return nil, fmt.Errorf("scan plan guardrail: %w", err)
|
||||
}
|
||||
g := ext.AgentGuardrail{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -422,6 +422,7 @@ func scanPlan(row planScanner) (ext.Plan, error) {
|
||||
var tags []string
|
||||
err := row.Scan(
|
||||
&model.ID,
|
||||
&model.GUID,
|
||||
&model.Title,
|
||||
&model.Description,
|
||||
&model.Status,
|
||||
@@ -467,9 +468,9 @@ func (db *DB) listPlansByQuery(ctx context.Context, query string, args ...any) (
|
||||
return plans, nil
|
||||
}
|
||||
|
||||
// canonicalPlanPair ensures the smaller UUID is always plan_a_id to prevent duplicates.
|
||||
func canonicalPlanPair(a, b uuid.UUID) (uuid.UUID, uuid.UUID) {
|
||||
if strings.Compare(a.String(), b.String()) <= 0 {
|
||||
// canonicalPlanPair ensures the smaller ID is always plan_a_id to prevent duplicates.
|
||||
func canonicalPlanPair(a, b int64) (int64, int64) {
|
||||
if a <= b {
|
||||
return a, b
|
||||
}
|
||||
return b, a
|
||||
|
||||
@@ -16,11 +16,11 @@ func (db *DB) CreateProject(ctx context.Context, name, description string) (thou
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
insert into projects (name, description)
|
||||
values ($1, $2)
|
||||
returning guid, name, description, created_at, last_active_at
|
||||
returning id, guid, name, description, created_at, last_active_at
|
||||
`, name, description)
|
||||
|
||||
var model generatedmodels.ModelPublicProjects
|
||||
if err := row.Scan(&model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt); err != nil {
|
||||
return thoughttypes.Project{}, fmt.Errorf("create project: %w", err)
|
||||
}
|
||||
return projectFromModel(model), nil
|
||||
@@ -46,7 +46,7 @@ func (db *DB) GetProject(ctx context.Context, nameOrID string) (thoughttypes.Pro
|
||||
|
||||
func (db *DB) getProjectByGUID(ctx context.Context, id uuid.UUID) (thoughttypes.Project, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select guid, name, description, created_at, last_active_at
|
||||
select id, guid, name, description, created_at, last_active_at
|
||||
from projects
|
||||
where guid = $1
|
||||
`, id)
|
||||
@@ -55,7 +55,7 @@ func (db *DB) getProjectByGUID(ctx context.Context, id uuid.UUID) (thoughttypes.
|
||||
|
||||
func (db *DB) getProjectByName(ctx context.Context, name string) (thoughttypes.Project, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select guid, name, description, created_at, last_active_at
|
||||
select id, guid, name, description, created_at, last_active_at
|
||||
from projects
|
||||
where name = $1
|
||||
`, name)
|
||||
@@ -64,7 +64,7 @@ func (db *DB) getProjectByName(ctx context.Context, name string) (thoughttypes.P
|
||||
|
||||
func scanProject(row pgx.Row) (thoughttypes.Project, error) {
|
||||
var model generatedmodels.ModelPublicProjects
|
||||
if err := row.Scan(&model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt); err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return thoughttypes.Project{}, err
|
||||
}
|
||||
@@ -75,10 +75,10 @@ func scanProject(row pgx.Row) (thoughttypes.Project, error) {
|
||||
|
||||
func (db *DB) ListProjects(ctx context.Context) ([]thoughttypes.ProjectSummary, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select p.guid, p.name, p.description, p.created_at, p.last_active_at, count(t.id) as thought_count
|
||||
select p.id, p.guid, p.name, p.description, p.created_at, p.last_active_at, count(t.id) as thought_count
|
||||
from projects p
|
||||
left join thoughts t on t.project_id = p.guid and t.archived_at is null
|
||||
group by p.guid, p.name, p.description, p.created_at, p.last_active_at
|
||||
left join thoughts t on t.project_id = p.id and t.archived_at is null
|
||||
group by p.id, p.guid, p.name, p.description, p.created_at, p.last_active_at
|
||||
order by p.last_active_at desc, p.created_at desc
|
||||
`)
|
||||
if err != nil {
|
||||
@@ -90,7 +90,7 @@ func (db *DB) ListProjects(ctx context.Context) ([]thoughttypes.ProjectSummary,
|
||||
for rows.Next() {
|
||||
var model generatedmodels.ModelPublicProjects
|
||||
var thoughtCount int
|
||||
if err := rows.Scan(&model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt, &thoughtCount); err != nil {
|
||||
if err := rows.Scan(&model.ID, &model.GUID, &model.Name, &model.Description, &model.CreatedAt, &model.LastActiveAt, &thoughtCount); err != nil {
|
||||
return nil, fmt.Errorf("scan project summary: %w", err)
|
||||
}
|
||||
projects = append(projects, thoughttypes.ProjectSummary{
|
||||
@@ -104,8 +104,8 @@ func (db *DB) ListProjects(ctx context.Context) ([]thoughttypes.ProjectSummary,
|
||||
return projects, nil
|
||||
}
|
||||
|
||||
func (db *DB) TouchProject(ctx context.Context, id uuid.UUID) error {
|
||||
tag, err := db.pool.Exec(ctx, `update projects set last_active_at = now() where guid = $1`, id)
|
||||
func (db *DB) TouchProject(ctx context.Context, id int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `update projects set last_active_at = now() where id = $1`, id)
|
||||
if err != nil {
|
||||
return fmt.Errorf("touch project: %w", err)
|
||||
}
|
||||
|
||||
@@ -5,8 +5,6 @@ import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/generatedmodels"
|
||||
ext "git.warky.dev/wdevs/amcs/internal/types"
|
||||
)
|
||||
@@ -20,21 +18,22 @@ func (db *DB) AddSkill(ctx context.Context, skill ext.AgentSkill) (ext.AgentSkil
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
insert into agent_skills (name, description, content, tags)
|
||||
values ($1, $2, $3, $4)
|
||||
returning id, created_at, updated_at
|
||||
returning id, guid, created_at, updated_at
|
||||
`, skill.Name, skill.Description, skill.Content, skill.Tags)
|
||||
|
||||
created := skill
|
||||
var model generatedmodels.ModelPublicAgentSkills
|
||||
if err := row.Scan(&model.ID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return ext.AgentSkill{}, fmt.Errorf("insert agent skill: %w", err)
|
||||
}
|
||||
created.ID = model.ID.UUID()
|
||||
created.ID = model.ID.Int64()
|
||||
created.GUID = model.GUID.UUID()
|
||||
created.CreatedAt = model.CreatedAt.Time()
|
||||
created.UpdatedAt = model.UpdatedAt.Time()
|
||||
return created, nil
|
||||
}
|
||||
|
||||
func (db *DB) RemoveSkill(ctx context.Context, id uuid.UUID) error {
|
||||
func (db *DB) RemoveSkill(ctx context.Context, id int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `delete from agent_skills where id = $1`, id)
|
||||
if err != nil {
|
||||
return fmt.Errorf("delete agent skill: %w", err)
|
||||
@@ -68,7 +67,7 @@ func (db *DB) ListSkills(ctx context.Context, tag string) ([]ext.AgentSkill, err
|
||||
return nil, fmt.Errorf("scan agent skill: %w", err)
|
||||
}
|
||||
s := ext.AgentSkill{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -84,7 +83,7 @@ func (db *DB) ListSkills(ctx context.Context, tag string) ([]ext.AgentSkill, err
|
||||
return skills, rows.Err()
|
||||
}
|
||||
|
||||
func (db *DB) GetSkill(ctx context.Context, id uuid.UUID) (ext.AgentSkill, error) {
|
||||
func (db *DB) GetSkill(ctx context.Context, id int64) (ext.AgentSkill, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select id, name, description, content, tags::text[], created_at, updated_at
|
||||
from agent_skills where id = $1
|
||||
@@ -96,7 +95,7 @@ func (db *DB) GetSkill(ctx context.Context, id uuid.UUID) (ext.AgentSkill, error
|
||||
return ext.AgentSkill{}, fmt.Errorf("get agent skill: %w", err)
|
||||
}
|
||||
s := ext.AgentSkill{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -122,21 +121,22 @@ func (db *DB) AddGuardrail(ctx context.Context, g ext.AgentGuardrail) (ext.Agent
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
insert into agent_guardrails (name, description, content, severity, tags)
|
||||
values ($1, $2, $3, $4, $5)
|
||||
returning id, created_at, updated_at
|
||||
returning id, guid, created_at, updated_at
|
||||
`, g.Name, g.Description, g.Content, g.Severity, g.Tags)
|
||||
|
||||
created := g
|
||||
var model generatedmodels.ModelPublicAgentGuardrails
|
||||
if err := row.Scan(&model.ID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return ext.AgentGuardrail{}, fmt.Errorf("insert agent guardrail: %w", err)
|
||||
}
|
||||
created.ID = model.ID.UUID()
|
||||
created.ID = model.ID.Int64()
|
||||
created.GUID = model.GUID.UUID()
|
||||
created.CreatedAt = model.CreatedAt.Time()
|
||||
created.UpdatedAt = model.UpdatedAt.Time()
|
||||
return created, nil
|
||||
}
|
||||
|
||||
func (db *DB) RemoveGuardrail(ctx context.Context, id uuid.UUID) error {
|
||||
func (db *DB) RemoveGuardrail(ctx context.Context, id int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `delete from agent_guardrails where id = $1`, id)
|
||||
if err != nil {
|
||||
return fmt.Errorf("delete agent guardrail: %w", err)
|
||||
@@ -180,7 +180,7 @@ func (db *DB) ListGuardrails(ctx context.Context, tag, severity string) ([]ext.A
|
||||
return nil, fmt.Errorf("scan agent guardrail: %w", err)
|
||||
}
|
||||
g := ext.AgentGuardrail{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -197,7 +197,7 @@ func (db *DB) ListGuardrails(ctx context.Context, tag, severity string) ([]ext.A
|
||||
return guardrails, rows.Err()
|
||||
}
|
||||
|
||||
func (db *DB) GetGuardrail(ctx context.Context, id uuid.UUID) (ext.AgentGuardrail, error) {
|
||||
func (db *DB) GetGuardrail(ctx context.Context, id int64) (ext.AgentGuardrail, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select id, name, description, content, severity, tags::text[], created_at, updated_at
|
||||
from agent_guardrails where id = $1
|
||||
@@ -209,7 +209,7 @@ func (db *DB) GetGuardrail(ctx context.Context, id uuid.UUID) (ext.AgentGuardrai
|
||||
return ext.AgentGuardrail{}, fmt.Errorf("get agent guardrail: %w", err)
|
||||
}
|
||||
g := ext.AgentGuardrail{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -226,7 +226,7 @@ func (db *DB) GetGuardrail(ctx context.Context, id uuid.UUID) (ext.AgentGuardrai
|
||||
|
||||
// Project Skills
|
||||
|
||||
func (db *DB) AddProjectSkill(ctx context.Context, projectID, skillID uuid.UUID) error {
|
||||
func (db *DB) AddProjectSkill(ctx context.Context, projectID, skillID int64) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into project_skills (project_id, skill_id)
|
||||
values ($1, $2)
|
||||
@@ -238,7 +238,7 @@ func (db *DB) AddProjectSkill(ctx context.Context, projectID, skillID uuid.UUID)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemoveProjectSkill(ctx context.Context, projectID, skillID uuid.UUID) error {
|
||||
func (db *DB) RemoveProjectSkill(ctx context.Context, projectID, skillID int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from project_skills where project_id = $1 and skill_id = $2
|
||||
`, projectID, skillID)
|
||||
@@ -251,7 +251,7 @@ func (db *DB) RemoveProjectSkill(ctx context.Context, projectID, skillID uuid.UU
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) ListProjectSkills(ctx context.Context, projectID uuid.UUID) ([]ext.AgentSkill, error) {
|
||||
func (db *DB) ListProjectSkills(ctx context.Context, projectID int64) ([]ext.AgentSkill, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select s.id, s.name, s.description, s.content, s.tags::text[], s.created_at, s.updated_at
|
||||
from agent_skills s
|
||||
@@ -272,7 +272,7 @@ func (db *DB) ListProjectSkills(ctx context.Context, projectID uuid.UUID) ([]ext
|
||||
return nil, fmt.Errorf("scan project skill: %w", err)
|
||||
}
|
||||
s := ext.AgentSkill{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
@@ -290,7 +290,7 @@ func (db *DB) ListProjectSkills(ctx context.Context, projectID uuid.UUID) ([]ext
|
||||
|
||||
// Project Guardrails
|
||||
|
||||
func (db *DB) AddProjectGuardrail(ctx context.Context, projectID, guardrailID uuid.UUID) error {
|
||||
func (db *DB) AddProjectGuardrail(ctx context.Context, projectID, guardrailID int64) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into project_guardrails (project_id, guardrail_id)
|
||||
values ($1, $2)
|
||||
@@ -302,7 +302,7 @@ func (db *DB) AddProjectGuardrail(ctx context.Context, projectID, guardrailID uu
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RemoveProjectGuardrail(ctx context.Context, projectID, guardrailID uuid.UUID) error {
|
||||
func (db *DB) RemoveProjectGuardrail(ctx context.Context, projectID, guardrailID int64) error {
|
||||
tag, err := db.pool.Exec(ctx, `
|
||||
delete from project_guardrails where project_id = $1 and guardrail_id = $2
|
||||
`, projectID, guardrailID)
|
||||
@@ -315,7 +315,7 @@ func (db *DB) RemoveProjectGuardrail(ctx context.Context, projectID, guardrailID
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) ListProjectGuardrails(ctx context.Context, projectID uuid.UUID) ([]ext.AgentGuardrail, error) {
|
||||
func (db *DB) ListProjectGuardrails(ctx context.Context, projectID int64) ([]ext.AgentGuardrail, error) {
|
||||
rows, err := db.pool.Query(ctx, `
|
||||
select g.id, g.name, g.description, g.content, g.severity, g.tags::text[], g.created_at, g.updated_at
|
||||
from agent_guardrails g
|
||||
@@ -336,7 +336,7 @@ func (db *DB) ListProjectGuardrails(ctx context.Context, projectID uuid.UUID) ([
|
||||
return nil, fmt.Errorf("scan project guardrail: %w", err)
|
||||
}
|
||||
g := ext.AgentGuardrail{
|
||||
ID: model.ID.UUID(),
|
||||
ID: model.ID.Int64(),
|
||||
Name: model.Name.String(),
|
||||
Description: model.Description.String(),
|
||||
Content: model.Content.String(),
|
||||
|
||||
@@ -33,12 +33,12 @@ func (db *DB) InsertThought(ctx context.Context, thought thoughttypes.Thought, e
|
||||
row := tx.QueryRow(ctx, `
|
||||
insert into thoughts (content, metadata, project_id)
|
||||
values ($1, $2::jsonb, $3)
|
||||
returning guid, created_at, updated_at
|
||||
returning id, guid, created_at, updated_at
|
||||
`, thought.Content, metadata, thought.ProjectID)
|
||||
|
||||
created := thought
|
||||
created.Embedding = nil
|
||||
if err := row.Scan(&created.ID, &created.CreatedAt, &created.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&created.ID, &created.GUID, &created.CreatedAt, &created.UpdatedAt); err != nil {
|
||||
return thoughttypes.Thought{}, fmt.Errorf("insert thought: %w", err)
|
||||
}
|
||||
|
||||
@@ -132,7 +132,7 @@ func (db *DB) ListThoughts(ctx context.Context, filter thoughttypes.ListFilter)
|
||||
}
|
||||
|
||||
query := `
|
||||
select guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
from thoughts
|
||||
`
|
||||
if len(conditions) > 0 {
|
||||
@@ -151,7 +151,7 @@ func (db *DB) ListThoughts(ctx context.Context, filter thoughttypes.ListFilter)
|
||||
thoughts := make([]thoughttypes.Thought, 0, filter.Limit)
|
||||
for rows.Next() {
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := rows.Scan(&model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := rows.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return nil, fmt.Errorf("scan listed thought: %w", err)
|
||||
}
|
||||
thought, err := thoughtFromModel(model)
|
||||
@@ -218,13 +218,13 @@ func (db *DB) Stats(ctx context.Context) (thoughttypes.ThoughtStats, error) {
|
||||
|
||||
func (db *DB) GetThought(ctx context.Context, id uuid.UUID) (thoughttypes.Thought, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
from thoughts
|
||||
where guid = $1
|
||||
`, id)
|
||||
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := row.Scan(&model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return thoughttypes.Thought{}, err
|
||||
}
|
||||
@@ -239,7 +239,30 @@ func (db *DB) GetThought(ctx context.Context, id uuid.UUID) (thoughttypes.Though
|
||||
return thought, nil
|
||||
}
|
||||
|
||||
func (db *DB) UpdateThought(ctx context.Context, id uuid.UUID, content string, embedding []float32, embeddingModel string, metadata thoughttypes.ThoughtMetadata, projectID *uuid.UUID) (thoughttypes.Thought, error) {
|
||||
func (db *DB) GetThoughtByID(ctx context.Context, id int64) (thoughttypes.Thought, error) {
|
||||
row := db.pool.QueryRow(ctx, `
|
||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
from thoughts
|
||||
where id = $1
|
||||
`, id)
|
||||
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
if err == pgx.ErrNoRows {
|
||||
return thoughttypes.Thought{}, err
|
||||
}
|
||||
return thoughttypes.Thought{}, fmt.Errorf("get thought by id: %w", err)
|
||||
}
|
||||
|
||||
thought, err := thoughtFromModel(model)
|
||||
if err != nil {
|
||||
return thoughttypes.Thought{}, fmt.Errorf("map thought: %w", err)
|
||||
}
|
||||
|
||||
return thought, nil
|
||||
}
|
||||
|
||||
func (db *DB) UpdateThought(ctx context.Context, id uuid.UUID, content string, embedding []float32, embeddingModel string, metadata thoughttypes.ThoughtMetadata, projectID *int64) (thoughttypes.Thought, error) {
|
||||
metadataBytes, err := json.Marshal(metadata)
|
||||
if err != nil {
|
||||
return thoughttypes.Thought{}, fmt.Errorf("marshal updated metadata: %w", err)
|
||||
@@ -271,7 +294,7 @@ func (db *DB) UpdateThought(ctx context.Context, id uuid.UUID, content string, e
|
||||
if len(embedding) > 0 && embeddingModel != "" {
|
||||
if _, err := tx.Exec(ctx, `
|
||||
insert into embeddings (thought_id, model, dim, embedding)
|
||||
values ($1, $2, $3, $4)
|
||||
select id, $2, $3, $4 from thoughts where guid = $1
|
||||
on conflict (thought_id, model) do update
|
||||
set embedding = excluded.embedding,
|
||||
dim = excluded.dim,
|
||||
@@ -288,7 +311,7 @@ func (db *DB) UpdateThought(ctx context.Context, id uuid.UUID, content string, e
|
||||
return db.GetThought(ctx, id)
|
||||
}
|
||||
|
||||
func (db *DB) UpdateThoughtMetadata(ctx context.Context, id uuid.UUID, metadata thoughttypes.ThoughtMetadata) (thoughttypes.Thought, error) {
|
||||
func (db *DB) UpdateThoughtMetadata(ctx context.Context, id int64, metadata thoughttypes.ThoughtMetadata) (thoughttypes.Thought, error) {
|
||||
metadataBytes, err := json.Marshal(metadata)
|
||||
if err != nil {
|
||||
return thoughttypes.Thought{}, fmt.Errorf("marshal updated metadata: %w", err)
|
||||
@@ -298,7 +321,7 @@ func (db *DB) UpdateThoughtMetadata(ctx context.Context, id uuid.UUID, metadata
|
||||
update thoughts
|
||||
set metadata = $2::jsonb,
|
||||
updated_at = now()
|
||||
where guid = $1
|
||||
where id = $1
|
||||
`, id, metadataBytes)
|
||||
if err != nil {
|
||||
return thoughttypes.Thought{}, fmt.Errorf("update thought metadata: %w", err)
|
||||
@@ -307,7 +330,7 @@ func (db *DB) UpdateThoughtMetadata(ctx context.Context, id uuid.UUID, metadata
|
||||
return thoughttypes.Thought{}, pgx.ErrNoRows
|
||||
}
|
||||
|
||||
return db.GetThought(ctx, id)
|
||||
return db.GetThoughtByID(ctx, id)
|
||||
}
|
||||
|
||||
func (db *DB) DeleteThought(ctx context.Context, id uuid.UUID) error {
|
||||
@@ -332,7 +355,7 @@ func (db *DB) ArchiveThought(ctx context.Context, id uuid.UUID) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) RecentThoughts(ctx context.Context, projectID *uuid.UUID, limit int, days int) ([]thoughttypes.Thought, error) {
|
||||
func (db *DB) RecentThoughts(ctx context.Context, projectID *int64, limit int, days int) ([]thoughttypes.Thought, error) {
|
||||
filter := thoughttypes.ListFilter{
|
||||
Limit: limit,
|
||||
ProjectID: projectID,
|
||||
@@ -342,7 +365,7 @@ func (db *DB) RecentThoughts(ctx context.Context, projectID *uuid.UUID, limit in
|
||||
return db.ListThoughts(ctx, filter)
|
||||
}
|
||||
|
||||
func (db *DB) ListThoughtsPendingMetadataRetry(ctx context.Context, limit int, projectID *uuid.UUID, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
func (db *DB) ListThoughtsPendingMetadataRetry(ctx context.Context, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
args := make([]any, 0, 4)
|
||||
conditions := []string{
|
||||
"(metadata->>'metadata_status' = 'pending' or metadata->>'metadata_status' = 'failed')",
|
||||
@@ -361,7 +384,7 @@ func (db *DB) ListThoughtsPendingMetadataRetry(ctx context.Context, limit int, p
|
||||
}
|
||||
|
||||
query := `
|
||||
select guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
from thoughts
|
||||
where ` + strings.Join(conditions, " and ")
|
||||
|
||||
@@ -376,12 +399,12 @@ func (db *DB) ListThoughtsPendingMetadataRetry(ctx context.Context, limit int, p
|
||||
|
||||
thoughts := make([]thoughttypes.Thought, 0, limit)
|
||||
for rows.Next() {
|
||||
var thought thoughttypes.Thought
|
||||
var metadataBytes []byte
|
||||
if err := rows.Scan(&thought.ID, &thought.Content, &metadataBytes, &thought.ProjectID, &thought.ArchivedAt, &thought.CreatedAt, &thought.UpdatedAt); err != nil {
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := rows.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return nil, fmt.Errorf("scan pending metadata retry thought: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(metadataBytes, &thought.Metadata); err != nil {
|
||||
thought, err := thoughtFromModel(model)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("decode pending metadata retry thought: %w", err)
|
||||
}
|
||||
thoughts = append(thoughts, thought)
|
||||
@@ -394,7 +417,7 @@ func (db *DB) ListThoughtsPendingMetadataRetry(ctx context.Context, limit int, p
|
||||
return thoughts, nil
|
||||
}
|
||||
|
||||
func (db *DB) SearchSimilarThoughts(ctx context.Context, embedding []float32, embeddingModel string, threshold float64, limit int, projectID *uuid.UUID, excludeID *uuid.UUID) ([]thoughttypes.SearchResult, error) {
|
||||
func (db *DB) SearchSimilarThoughts(ctx context.Context, embedding []float32, embeddingModel string, threshold float64, limit int, projectID *int64, excludeID *uuid.UUID) ([]thoughttypes.SearchResult, error) {
|
||||
args := []any{pgvector.NewVector(embedding), threshold, embeddingModel}
|
||||
conditions := []string{
|
||||
"t.archived_at is null",
|
||||
@@ -412,9 +435,9 @@ func (db *DB) SearchSimilarThoughts(ctx context.Context, embedding []float32, em
|
||||
args = append(args, limit)
|
||||
|
||||
query := `
|
||||
select t.guid, t.content, t.metadata, 1 - (e.embedding <=> $1) as similarity, t.created_at
|
||||
select t.id, t.content, t.metadata, 1 - (e.embedding <=> $1) as similarity, t.created_at
|
||||
from thoughts t
|
||||
join embeddings e on e.thought_id = t.guid
|
||||
join embeddings e on e.thought_id = t.id
|
||||
where ` + strings.Join(conditions, " and ") + fmt.Sprintf(`
|
||||
order by e.embedding <=> $1
|
||||
limit $%d`, len(args))
|
||||
@@ -443,7 +466,7 @@ func (db *DB) SearchSimilarThoughts(ctx context.Context, embedding []float32, em
|
||||
return results, nil
|
||||
}
|
||||
|
||||
func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID *uuid.UUID) (bool, error) {
|
||||
func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID *int64) (bool, error) {
|
||||
args := []any{model}
|
||||
conditions := []string{
|
||||
"e.model = $1",
|
||||
@@ -454,7 +477,7 @@ func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID
|
||||
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
||||
}
|
||||
|
||||
query := `select exists(select 1 from embeddings e join thoughts t on t.guid = e.thought_id where ` +
|
||||
query := `select exists(select 1 from embeddings e join thoughts t on t.id = e.thought_id where ` +
|
||||
strings.Join(conditions, " and ") + `)`
|
||||
|
||||
var exists bool
|
||||
@@ -464,7 +487,7 @@ func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID
|
||||
return exists, nil
|
||||
}
|
||||
|
||||
func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, limit int, projectID *uuid.UUID, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
args := []any{model}
|
||||
conditions := []string{"e.id is null"}
|
||||
|
||||
@@ -482,9 +505,9 @@ func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, li
|
||||
args = append(args, limit)
|
||||
|
||||
query := `
|
||||
select t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at
|
||||
select t.id, t.guid, t.content, t.metadata, t.project_id, t.archived_at, t.created_at, t.updated_at
|
||||
from thoughts t
|
||||
left join embeddings e on e.thought_id = t.guid and e.model = $1
|
||||
left join embeddings e on e.thought_id = t.id and e.model = $1
|
||||
where ` + strings.Join(conditions, " and ") + `
|
||||
order by t.created_at asc
|
||||
limit $` + fmt.Sprintf("%d", len(args))
|
||||
@@ -497,12 +520,12 @@ func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, li
|
||||
|
||||
thoughts := make([]thoughttypes.Thought, 0, limit)
|
||||
for rows.Next() {
|
||||
var thought thoughttypes.Thought
|
||||
var metadataBytes []byte
|
||||
if err := rows.Scan(&thought.ID, &thought.Content, &metadataBytes, &thought.ProjectID, &thought.ArchivedAt, &thought.CreatedAt, &thought.UpdatedAt); err != nil {
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := rows.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return nil, fmt.Errorf("scan missing-embedding thought: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(metadataBytes, &thought.Metadata); err != nil {
|
||||
thought, err := thoughtFromModel(model)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("decode missing-embedding metadata: %w", err)
|
||||
}
|
||||
thoughts = append(thoughts, thought)
|
||||
@@ -513,7 +536,7 @@ func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, li
|
||||
return thoughts, nil
|
||||
}
|
||||
|
||||
func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, projectID *uuid.UUID, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||
args := make([]any, 0, 3)
|
||||
conditions := make([]string, 0, 4)
|
||||
|
||||
@@ -531,7 +554,7 @@ func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, pro
|
||||
args = append(args, limit)
|
||||
|
||||
query := `
|
||||
select guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||
from thoughts
|
||||
`
|
||||
if len(conditions) > 0 {
|
||||
@@ -547,12 +570,12 @@ func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, pro
|
||||
|
||||
thoughts := make([]thoughttypes.Thought, 0, limit)
|
||||
for rows.Next() {
|
||||
var thought thoughttypes.Thought
|
||||
var metadataBytes []byte
|
||||
if err := rows.Scan(&thought.ID, &thought.Content, &metadataBytes, &thought.ProjectID, &thought.ArchivedAt, &thought.CreatedAt, &thought.UpdatedAt); err != nil {
|
||||
var model generatedmodels.ModelPublicThoughts
|
||||
if err := rows.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||
return nil, fmt.Errorf("scan metadata-reparse thought: %w", err)
|
||||
}
|
||||
if err := json.Unmarshal(metadataBytes, &thought.Metadata); err != nil {
|
||||
thought, err := thoughtFromModel(model)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("decode metadata-reparse thought metadata: %w", err)
|
||||
}
|
||||
thoughts = append(thoughts, thought)
|
||||
@@ -564,7 +587,7 @@ func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, pro
|
||||
return thoughts, nil
|
||||
}
|
||||
|
||||
func (db *DB) UpsertEmbedding(ctx context.Context, thoughtID uuid.UUID, model string, embedding []float32) error {
|
||||
func (db *DB) UpsertEmbedding(ctx context.Context, thoughtID int64, model string, embedding []float32) error {
|
||||
_, err := db.pool.Exec(ctx, `
|
||||
insert into embeddings (thought_id, model, dim, embedding)
|
||||
values ($1, $2, $3, $4)
|
||||
@@ -579,7 +602,7 @@ func (db *DB) UpsertEmbedding(ctx context.Context, thoughtID uuid.UUID, model st
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DB) SearchThoughtsText(ctx context.Context, query string, limit int, projectID *uuid.UUID, excludeID *uuid.UUID) ([]thoughttypes.SearchResult, error) {
|
||||
func (db *DB) SearchThoughtsText(ctx context.Context, query string, limit int, projectID *int64, excludeID *uuid.UUID) ([]thoughttypes.SearchResult, error) {
|
||||
args := []any{query}
|
||||
conditions := []string{
|
||||
"t.archived_at is null",
|
||||
@@ -596,11 +619,11 @@ func (db *DB) SearchThoughtsText(ctx context.Context, query string, limit int, p
|
||||
args = append(args, limit)
|
||||
|
||||
q := `
|
||||
select t.guid, t.content, t.metadata,
|
||||
select t.id, t.content, t.metadata,
|
||||
ts_rank_cd(to_tsvector('simple', t.content) || to_tsvector('simple', coalesce(p.name, '')), websearch_to_tsquery('simple', $1)) as similarity,
|
||||
t.created_at
|
||||
from thoughts t
|
||||
left join projects p on t.project_id = p.guid
|
||||
left join projects p on t.project_id = p.id
|
||||
where ` + strings.Join(conditions, " and ") + `
|
||||
order by similarity desc
|
||||
limit $` + fmt.Sprintf("%d", len(args))
|
||||
|
||||
@@ -2,11 +2,11 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
"golang.org/x/sync/semaphore"
|
||||
|
||||
@@ -53,11 +53,12 @@ func NewBackfillTool(db *store.DB, embeddings *ai.EmbeddingRunner, sessions *ses
|
||||
|
||||
// QueueThought queues a single thought for background embedding generation.
|
||||
// It is used by capture when the embedding provider is temporarily unavailable.
|
||||
func (t *BackfillTool) QueueThought(ctx context.Context, id uuid.UUID, content string) {
|
||||
func (t *BackfillTool) QueueThought(ctx context.Context, id int64, content string) {
|
||||
go func() {
|
||||
started := time.Now()
|
||||
idStr := fmt.Sprint(id)
|
||||
t.logger.Info("background embedding started",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", t.embeddings.PrimaryModel()),
|
||||
)
|
||||
@@ -65,7 +66,7 @@ func (t *BackfillTool) QueueThought(ctx context.Context, id uuid.UUID, content s
|
||||
result, err := t.embeddings.Embed(ctx, content)
|
||||
if err != nil {
|
||||
t.logger.Warn("background embedding error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", t.embeddings.PrimaryModel()),
|
||||
slog.String("stage", "embed"),
|
||||
@@ -76,7 +77,7 @@ func (t *BackfillTool) QueueThought(ctx context.Context, id uuid.UUID, content s
|
||||
}
|
||||
if err := t.store.UpsertEmbedding(ctx, id, result.Model, result.Vector); err != nil {
|
||||
t.logger.Warn("background embedding error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", result.Model),
|
||||
slog.String("stage", "upsert"),
|
||||
@@ -86,7 +87,7 @@ func (t *BackfillTool) QueueThought(ctx context.Context, id uuid.UUID, content s
|
||||
return
|
||||
}
|
||||
t.logger.Info("background embedding complete",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", t.embeddings.PrimaryProvider()),
|
||||
slog.String("model", result.Model),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
@@ -105,9 +106,9 @@ func (t *BackfillTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in
|
||||
return nil, BackfillOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
primaryModel := t.embeddings.PrimaryModel()
|
||||
@@ -140,24 +141,25 @@ func (t *BackfillTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in
|
||||
break
|
||||
}
|
||||
wg.Add(1)
|
||||
go func(id uuid.UUID, content string) {
|
||||
go func(id int64, content string) {
|
||||
defer wg.Done()
|
||||
defer sem.Release(1)
|
||||
|
||||
idStr := fmt.Sprint(id)
|
||||
result, embedErr := t.embeddings.Embed(ctx, content)
|
||||
if embedErr != nil {
|
||||
mu.Lock()
|
||||
out.Failures = append(out.Failures, BackfillFailure{ID: id.String(), Error: embedErr.Error()})
|
||||
out.Failures = append(out.Failures, BackfillFailure{ID: idStr, Error: embedErr.Error()})
|
||||
mu.Unlock()
|
||||
t.logger.Warn("backfill embed failed", slog.String("thought_id", id.String()), slog.String("error", embedErr.Error()))
|
||||
t.logger.Warn("backfill embed failed", slog.String("thought_id", idStr), slog.String("error", embedErr.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
if upsertErr := t.store.UpsertEmbedding(ctx, id, result.Model, result.Vector); upsertErr != nil {
|
||||
mu.Lock()
|
||||
out.Failures = append(out.Failures, BackfillFailure{ID: id.String(), Error: upsertErr.Error()})
|
||||
out.Failures = append(out.Failures, BackfillFailure{ID: idStr, Error: upsertErr.Error()})
|
||||
mu.Unlock()
|
||||
t.logger.Warn("backfill upsert failed", slog.String("thought_id", id.String()), slog.String("error", upsertErr.Error()))
|
||||
t.logger.Warn("backfill upsert failed", slog.String("thought_id", idStr), slog.String("error", upsertErr.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/ai"
|
||||
@@ -17,13 +16,13 @@ import (
|
||||
|
||||
// EmbeddingQueuer queues a thought for background embedding generation.
|
||||
type EmbeddingQueuer interface {
|
||||
QueueThought(ctx context.Context, id uuid.UUID, content string)
|
||||
QueueThought(ctx context.Context, id int64, content string)
|
||||
}
|
||||
|
||||
// MetadataQueuer queues a thought for background metadata retry. Both
|
||||
// MetadataRetryer and EnrichmentRetryer satisfy this.
|
||||
type MetadataQueuer interface {
|
||||
QueueThought(id uuid.UUID)
|
||||
QueueThought(id int64)
|
||||
}
|
||||
|
||||
type CaptureTool struct {
|
||||
@@ -66,7 +65,7 @@ func (t *CaptureTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in C
|
||||
Metadata: rawMetadata,
|
||||
}
|
||||
if project != nil {
|
||||
thought.ProjectID = &project.ID
|
||||
thought.ProjectID = &project.NumericID
|
||||
}
|
||||
|
||||
created, err := t.store.InsertThought(ctx, thought, t.embeddings.PrimaryModel())
|
||||
@@ -74,7 +73,7 @@ func (t *CaptureTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in C
|
||||
return nil, CaptureOutput{}, err
|
||||
}
|
||||
if project != nil {
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
if t.retryer != nil {
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/session"
|
||||
@@ -64,7 +63,7 @@ func (t *ChatHistoryTool) SaveChatHistory(ctx context.Context, req *mcp.CallTool
|
||||
h.Metadata = map[string]any{}
|
||||
}
|
||||
if project != nil {
|
||||
h.ProjectID = &project.ID
|
||||
h.ProjectID = &project.NumericID
|
||||
}
|
||||
|
||||
saved, err := t.store.SaveChatHistory(ctx, h)
|
||||
@@ -77,7 +76,7 @@ func (t *ChatHistoryTool) SaveChatHistory(ctx context.Context, req *mcp.CallTool
|
||||
// get_chat_history
|
||||
|
||||
type GetChatHistoryInput struct {
|
||||
ID string `json:"id,omitempty" jsonschema:"UUID of the saved chat history"`
|
||||
ID string `json:"id,omitempty" jsonschema:"numeric id of the saved chat history"`
|
||||
SessionID string `json:"session_id,omitempty" jsonschema:"original session_id — returns the most recent history for that session"`
|
||||
}
|
||||
|
||||
@@ -91,9 +90,9 @@ func (t *ChatHistoryTool) GetChatHistory(ctx context.Context, _ *mcp.CallToolReq
|
||||
}
|
||||
|
||||
if in.ID != "" {
|
||||
id, err := uuid.Parse(in.ID)
|
||||
id, err := parseID(in.ID)
|
||||
if err != nil {
|
||||
return nil, GetChatHistoryOutput{}, errInvalidField("id", "invalid id", "must be a valid UUID")
|
||||
return nil, GetChatHistoryOutput{}, errInvalidField("id", "invalid id", "must be a valid numeric id")
|
||||
}
|
||||
h, found, err := t.store.GetChatHistory(ctx, id)
|
||||
if err != nil {
|
||||
@@ -145,7 +144,7 @@ func (t *ChatHistoryTool) ListChatHistories(ctx context.Context, req *mcp.CallTo
|
||||
return nil, ListChatHistoriesOutput{}, err
|
||||
}
|
||||
if project != nil {
|
||||
filter.ProjectID = &project.ID
|
||||
filter.ProjectID = &project.NumericID
|
||||
}
|
||||
}
|
||||
|
||||
@@ -162,7 +161,7 @@ func (t *ChatHistoryTool) ListChatHistories(ctx context.Context, req *mcp.CallTo
|
||||
// delete_chat_history
|
||||
|
||||
type DeleteChatHistoryInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"UUID of the chat history to delete"`
|
||||
ID string `json:"id" jsonschema:"numeric id of the chat history to delete"`
|
||||
}
|
||||
|
||||
type DeleteChatHistoryOutput struct {
|
||||
@@ -170,7 +169,11 @@ type DeleteChatHistoryOutput struct {
|
||||
}
|
||||
|
||||
func (t *ChatHistoryTool) DeleteChatHistory(ctx context.Context, _ *mcp.CallToolRequest, in DeleteChatHistoryInput) (*mcp.CallToolResult, DeleteChatHistoryOutput, error) {
|
||||
deleted, err := t.store.DeleteChatHistory(ctx, in.ID)
|
||||
id, err := parseID(in.ID)
|
||||
if err != nil {
|
||||
return nil, DeleteChatHistoryOutput{}, errInvalidField("id", "invalid id", "must be a valid numeric id")
|
||||
}
|
||||
deleted, err := t.store.DeleteChatHistory(ctx, id)
|
||||
if err != nil {
|
||||
return nil, DeleteChatHistoryOutput{}, err
|
||||
}
|
||||
|
||||
@@ -52,7 +52,7 @@ func (t *ContextTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in P
|
||||
}
|
||||
|
||||
limit := normalizeLimit(in.Limit, t.search)
|
||||
recent, err := t.store.RecentThoughts(ctx, &project.ID, limit, 0)
|
||||
recent, err := t.store.RecentThoughts(ctx, &project.NumericID, limit, 0)
|
||||
if err != nil {
|
||||
return nil, ProjectContextOutput{}, err
|
||||
}
|
||||
@@ -60,7 +60,7 @@ func (t *ContextTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in P
|
||||
items := make([]ContextItem, 0, limit*2)
|
||||
seen := map[string]struct{}{}
|
||||
for _, thought := range recent {
|
||||
key := thought.ID.String()
|
||||
key := fmt.Sprint(thought.ID)
|
||||
seen[key] = struct{}{}
|
||||
items = append(items, ContextItem{
|
||||
ID: key,
|
||||
@@ -72,12 +72,12 @@ func (t *ContextTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in P
|
||||
|
||||
query := strings.TrimSpace(in.Query)
|
||||
if query != "" {
|
||||
semantic, err := semanticSearch(ctx, t.store, t.embeddings, t.search, query, limit, t.search.DefaultThreshold, &project.ID, nil)
|
||||
semantic, err := semanticSearch(ctx, t.store, t.embeddings, t.search, query, limit, t.search.DefaultThreshold, &project.NumericID, nil)
|
||||
if err != nil {
|
||||
return nil, ProjectContextOutput{}, err
|
||||
}
|
||||
for _, result := range semantic {
|
||||
key := result.ID.String()
|
||||
key := fmt.Sprint(result.ID)
|
||||
if _, ok := seen[key]; ok {
|
||||
continue
|
||||
}
|
||||
@@ -97,7 +97,7 @@ func (t *ContextTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in P
|
||||
lines = append(lines, thoughtContextLine(i, item.Content, item.Metadata, item.Similarity))
|
||||
}
|
||||
contextBlock := formatContextBlock(fmt.Sprintf("Project context for %s", project.Name), lines)
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
|
||||
return nil, ProjectContextOutput{
|
||||
Project: *project,
|
||||
|
||||
@@ -2,11 +2,11 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
"golang.org/x/sync/semaphore"
|
||||
|
||||
@@ -89,18 +89,19 @@ func (t *RetryEnrichmentTool) Handle(ctx context.Context, req *mcp.CallToolReque
|
||||
return t.retryer.Handle(ctx, req, in)
|
||||
}
|
||||
|
||||
func (r *EnrichmentRetryer) QueueThought(id uuid.UUID) {
|
||||
func (r *EnrichmentRetryer) QueueThought(id int64) {
|
||||
go func() {
|
||||
started := time.Now()
|
||||
idStr := fmt.Sprint(id)
|
||||
r.logger.Info("background metadata started",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
)
|
||||
updated, err := r.retryOne(r.backgroundCtx, id)
|
||||
if err != nil {
|
||||
r.logger.Warn("background metadata error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
@@ -109,7 +110,7 @@ func (r *EnrichmentRetryer) QueueThought(id uuid.UUID) {
|
||||
return
|
||||
}
|
||||
r.logger.Info("background metadata complete",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
slog.Bool("updated", updated),
|
||||
@@ -129,9 +130,9 @@ func (r *EnrichmentRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest
|
||||
return nil, RetryEnrichmentOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
thoughts, err := r.store.ListThoughtsPendingMetadataRetry(ctx, limit, projectID, in.IncludeArchived, in.OlderThanDays)
|
||||
@@ -168,7 +169,7 @@ func (r *EnrichmentRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest
|
||||
updated, err := r.retryOne(ctx, thought.ID)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
out.Failures = append(out.Failures, RetryEnrichmentFailure{ID: thought.ID.String(), Error: err.Error()})
|
||||
out.Failures = append(out.Failures, RetryEnrichmentFailure{ID: fmt.Sprint(thought.ID), Error: err.Error()})
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
@@ -191,8 +192,8 @@ func (r *EnrichmentRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest
|
||||
return nil, out, nil
|
||||
}
|
||||
|
||||
func (r *EnrichmentRetryer) retryOne(ctx context.Context, id uuid.UUID) (bool, error) {
|
||||
thought, err := r.store.GetThought(ctx, id)
|
||||
func (r *EnrichmentRetryer) retryOne(ctx context.Context, id int64) (bool, error) {
|
||||
thought, err := r.store.GetThoughtByID(ctx, id)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
@@ -5,6 +5,7 @@ import (
|
||||
"crypto/sha256"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
@@ -148,7 +149,7 @@ func (t *FilesTool) Upload(ctx context.Context, req *mcp.CallToolRequest, in Upl
|
||||
return nil, UploadFileOutput{}, err
|
||||
}
|
||||
|
||||
uri := fileURIPrefix + out.File.ID.String()
|
||||
uri := fileURIPrefix + fmt.Sprint(out.File.ID)
|
||||
return nil, UploadFileOutput{File: out.File, URI: uri}, nil
|
||||
}
|
||||
|
||||
@@ -247,7 +248,7 @@ func (t *FilesTool) Load(ctx context.Context, _ *mcp.CallToolRequest, in LoadFil
|
||||
return nil, LoadFileOutput{}, err
|
||||
}
|
||||
|
||||
uri := fileURIPrefix + file.ID.String()
|
||||
uri := fileURIPrefix + fmt.Sprint(file.ID)
|
||||
result := &mcp.CallToolResult{
|
||||
Content: []mcp.Content{
|
||||
&mcp.EmbeddedResource{
|
||||
@@ -294,7 +295,7 @@ func (t *FilesTool) List(ctx context.Context, req *mcp.CallToolRequest, in ListF
|
||||
return nil, ListFilesOutput{}, err
|
||||
}
|
||||
|
||||
var thoughtID *uuid.UUID
|
||||
var thoughtID *int64
|
||||
if rawThoughtID := strings.TrimSpace(in.ThoughtID); rawThoughtID != "" {
|
||||
parsedThoughtID, err := parseUUID(rawThoughtID)
|
||||
if err != nil {
|
||||
@@ -304,12 +305,12 @@ func (t *FilesTool) List(ctx context.Context, req *mcp.CallToolRequest, in ListF
|
||||
if err != nil {
|
||||
return nil, ListFilesOutput{}, err
|
||||
}
|
||||
thoughtID = &parsedThoughtID
|
||||
if project != nil && thought.ProjectID != nil && *thought.ProjectID != project.ID {
|
||||
thoughtID = &thought.ID
|
||||
if project != nil && thought.ProjectID != nil && *thought.ProjectID != project.NumericID {
|
||||
return nil, ListFilesOutput{}, errInvalidInput("project does not match the linked thought's project")
|
||||
}
|
||||
if project == nil && thought.ProjectID != nil {
|
||||
project = &thoughttypes.Project{ID: *thought.ProjectID}
|
||||
project = &thoughttypes.Project{NumericID: *thought.ProjectID}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -323,7 +324,7 @@ func (t *FilesTool) List(ctx context.Context, req *mcp.CallToolRequest, in ListF
|
||||
return nil, ListFilesOutput{}, err
|
||||
}
|
||||
if project != nil {
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
return nil, ListFilesOutput{Files: files}, nil
|
||||
@@ -343,7 +344,7 @@ func (t *FilesTool) SaveDecoded(ctx context.Context, req *mcp.CallToolRequest, i
|
||||
return SaveFileOutput{}, err
|
||||
}
|
||||
|
||||
var thoughtID *uuid.UUID
|
||||
var thoughtNumericID *int64
|
||||
var projectID = projectIDPtr(project)
|
||||
if rawThoughtID := strings.TrimSpace(in.ThoughtID); rawThoughtID != "" {
|
||||
parsedThoughtID, err := parseUUID(rawThoughtID)
|
||||
@@ -354,9 +355,9 @@ func (t *FilesTool) SaveDecoded(ctx context.Context, req *mcp.CallToolRequest, i
|
||||
if err != nil {
|
||||
return SaveFileOutput{}, err
|
||||
}
|
||||
thoughtID = &parsedThoughtID
|
||||
thoughtNumericID = &thought.ID
|
||||
projectID = thought.ProjectID
|
||||
if project != nil && thought.ProjectID != nil && *thought.ProjectID != project.ID {
|
||||
if project != nil && thought.ProjectID != nil && *thought.ProjectID != project.NumericID {
|
||||
return SaveFileOutput{}, errInvalidInput("project does not match the linked thought's project")
|
||||
}
|
||||
}
|
||||
@@ -374,9 +375,7 @@ func (t *FilesTool) SaveDecoded(ctx context.Context, req *mcp.CallToolRequest, i
|
||||
SHA256: hex.EncodeToString(sum[:]),
|
||||
Content: in.Content,
|
||||
ProjectID: projectID,
|
||||
}
|
||||
if thoughtID != nil {
|
||||
file.ThoughtID = thoughtID
|
||||
ThoughtID: thoughtNumericID,
|
||||
}
|
||||
|
||||
created, err := t.store.InsertStoredFile(ctx, file)
|
||||
@@ -398,7 +397,7 @@ func (t *FilesTool) SaveDecoded(ctx context.Context, req *mcp.CallToolRequest, i
|
||||
|
||||
func thoughtAttachmentFromFile(file thoughttypes.StoredFile) thoughttypes.ThoughtAttachment {
|
||||
return thoughttypes.ThoughtAttachment{
|
||||
FileID: file.ID,
|
||||
FileID: file.GUID,
|
||||
Name: file.Name,
|
||||
MediaType: file.MediaType,
|
||||
Kind: file.Kind,
|
||||
@@ -498,11 +497,11 @@ func normalizeFileKind(explicit string, mediaType string) string {
|
||||
}
|
||||
}
|
||||
|
||||
func projectIDPtr(project *thoughttypes.Project) *uuid.UUID {
|
||||
func projectIDPtr(project *thoughttypes.Project) *int64 {
|
||||
if project == nil {
|
||||
return nil
|
||||
}
|
||||
return &project.ID
|
||||
return &project.NumericID
|
||||
}
|
||||
|
||||
func normalizeFileLimit(limit int) int {
|
||||
|
||||
@@ -3,6 +3,7 @@ package tools
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
@@ -34,6 +35,14 @@ func parseUUID(id string) (uuid.UUID, error) {
|
||||
return parsed, nil
|
||||
}
|
||||
|
||||
func parseID(id string) (int64, error) {
|
||||
n, err := strconv.ParseInt(strings.TrimSpace(id), 10, 64)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("invalid numeric id %q", id)
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
|
||||
func sessionID(req *mcp.CallToolRequest) (string, error) {
|
||||
if req == nil || req.Session == nil || req.Session.ID() == "" {
|
||||
return "", newMCPError(
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/config"
|
||||
@@ -31,11 +30,11 @@ type AddLearningInput struct {
|
||||
SourceType string `json:"source_type,omitempty"`
|
||||
SourceRef string `json:"source_ref,omitempty"`
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id; falls back to active session project"`
|
||||
RelatedThoughtID *uuid.UUID `json:"related_thought_id,omitempty"`
|
||||
RelatedSkillID *uuid.UUID `json:"related_skill_id,omitempty"`
|
||||
RelatedThoughtID *int64 `json:"related_thought_id,omitempty"`
|
||||
RelatedSkillID *int64 `json:"related_skill_id,omitempty"`
|
||||
ReviewedBy *string `json:"reviewed_by,omitempty"`
|
||||
DuplicateOfLearningID *uuid.UUID `json:"duplicate_of_learning_id,omitempty"`
|
||||
SupersedesLearningID *uuid.UUID `json:"supersedes_learning_id,omitempty"`
|
||||
DuplicateOfLearningID *int64 `json:"duplicate_of_learning_id,omitempty"`
|
||||
SupersedesLearningID *int64 `json:"supersedes_learning_id,omitempty"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
}
|
||||
|
||||
@@ -44,7 +43,7 @@ type AddLearningOutput struct {
|
||||
}
|
||||
|
||||
type GetLearningInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"learning id"`
|
||||
ID int64 `json:"id" jsonschema:"learning id"`
|
||||
}
|
||||
|
||||
type GetLearningOutput struct {
|
||||
@@ -105,7 +104,7 @@ func (t *LearningsTool) Add(ctx context.Context, req *mcp.CallToolRequest, in Ad
|
||||
learning.ActionRequired = *in.ActionRequired
|
||||
}
|
||||
if project != nil {
|
||||
learning.ProjectID = &project.ID
|
||||
learning.ProjectID = &project.NumericID
|
||||
}
|
||||
|
||||
created, err := t.store.CreateLearning(ctx, learning)
|
||||
@@ -147,7 +146,7 @@ func (t *LearningsTool) List(ctx context.Context, req *mcp.CallToolRequest, in L
|
||||
Query: strings.TrimSpace(in.Query),
|
||||
}
|
||||
if project != nil {
|
||||
filter.ProjectID = &project.ID
|
||||
filter.ProjectID = &project.NumericID
|
||||
}
|
||||
|
||||
items, err := t.store.ListLearnings(ctx, filter)
|
||||
|
||||
@@ -5,7 +5,6 @@ import (
|
||||
"testing"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/mcperrors"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func TestLearningsAddRequiresSummary(t *testing.T) {
|
||||
@@ -37,7 +36,7 @@ func TestLearningsMethodsRequireConfiguredStore(t *testing.T) {
|
||||
})
|
||||
|
||||
t.Run("get", func(t *testing.T) {
|
||||
_, _, err := tool.Get(context.Background(), nil, GetLearningInput{ID: uuid.New()})
|
||||
_, _, err := tool.Get(context.Background(), nil, GetLearningInput{ID: 0})
|
||||
if err == nil {
|
||||
t.Fatal("Get() error = nil, want error")
|
||||
}
|
||||
|
||||
@@ -2,6 +2,7 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
@@ -97,9 +98,9 @@ func (t *LinksTool) Related(ctx context.Context, _ *mcp.CallToolRequest, in Rela
|
||||
}
|
||||
|
||||
related := make([]RelatedThought, 0, len(linked)+t.search.DefaultLimit)
|
||||
seen := map[string]struct{}{thought.ID.String(): {}}
|
||||
seen := map[string]struct{}{fmt.Sprint(thought.ID): {}}
|
||||
for _, item := range linked {
|
||||
key := item.Thought.ID.String()
|
||||
key := fmt.Sprint(item.Thought.ID)
|
||||
seen[key] = struct{}{}
|
||||
related = append(related, RelatedThought{
|
||||
ID: key,
|
||||
@@ -117,12 +118,12 @@ func (t *LinksTool) Related(ctx context.Context, _ *mcp.CallToolRequest, in Rela
|
||||
}
|
||||
|
||||
if includeSemantic {
|
||||
semantic, err := semanticSearch(ctx, t.store, t.embeddings, t.search, thought.Content, t.search.DefaultLimit, t.search.DefaultThreshold, thought.ProjectID, &thought.ID)
|
||||
semantic, err := semanticSearch(ctx, t.store, t.embeddings, t.search, thought.Content, t.search.DefaultLimit, t.search.DefaultThreshold, thought.ProjectID, &thought.GUID)
|
||||
if err != nil {
|
||||
return nil, RelatedOutput{}, err
|
||||
}
|
||||
for _, item := range semantic {
|
||||
key := item.ID.String()
|
||||
key := fmt.Sprint(item.ID)
|
||||
if _, ok := seen[key]; ok {
|
||||
continue
|
||||
}
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/config"
|
||||
@@ -43,9 +42,9 @@ func (t *ListTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in List
|
||||
return nil, ListOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
thoughts, err := t.store.ListThoughts(ctx, thoughttypes.ListFilter{
|
||||
@@ -61,7 +60,7 @@ func (t *ListTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in List
|
||||
return nil, ListOutput{}, err
|
||||
}
|
||||
if project != nil {
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
return nil, ListOutput{Thoughts: thoughts}, nil
|
||||
|
||||
@@ -2,11 +2,11 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
"golang.org/x/sync/semaphore"
|
||||
|
||||
@@ -37,18 +37,18 @@ type RetryMetadataTool struct {
|
||||
|
||||
type RetryLocker struct {
|
||||
mu sync.Mutex
|
||||
locks map[uuid.UUID]time.Time
|
||||
locks map[int64]time.Time
|
||||
}
|
||||
|
||||
func NewRetryLocker() *RetryLocker {
|
||||
return &RetryLocker{locks: map[uuid.UUID]time.Time{}}
|
||||
return &RetryLocker{locks: map[int64]time.Time{}}
|
||||
}
|
||||
|
||||
func (l *RetryLocker) Acquire(id uuid.UUID, ttl time.Duration) bool {
|
||||
func (l *RetryLocker) Acquire(id int64, ttl time.Duration) bool {
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
if l.locks == nil {
|
||||
l.locks = map[uuid.UUID]time.Time{}
|
||||
l.locks = map[int64]time.Time{}
|
||||
}
|
||||
now := time.Now()
|
||||
if exp, ok := l.locks[id]; ok && exp.After(now) {
|
||||
@@ -58,7 +58,7 @@ func (l *RetryLocker) Acquire(id uuid.UUID, ttl time.Duration) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (l *RetryLocker) Release(id uuid.UUID) {
|
||||
func (l *RetryLocker) Release(id int64) {
|
||||
l.mu.Lock()
|
||||
defer l.mu.Unlock()
|
||||
delete(l.locks, id)
|
||||
@@ -111,23 +111,24 @@ func (t *RetryMetadataTool) Handle(ctx context.Context, req *mcp.CallToolRequest
|
||||
return t.retryer.Handle(ctx, req, in)
|
||||
}
|
||||
|
||||
func (r *MetadataRetryer) QueueThought(id uuid.UUID) {
|
||||
func (r *MetadataRetryer) QueueThought(id int64) {
|
||||
go func() {
|
||||
started := time.Now()
|
||||
idStr := fmt.Sprint(id)
|
||||
if !r.lock.Acquire(id, 15*time.Minute) {
|
||||
return
|
||||
}
|
||||
defer r.lock.Release(id)
|
||||
|
||||
r.logger.Info("background metadata started",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
)
|
||||
updated, err := r.retryOne(r.backgroundCtx, id)
|
||||
if err != nil {
|
||||
r.logger.Warn("background metadata error",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
slog.Duration("duration", time.Since(started)),
|
||||
@@ -136,7 +137,7 @@ func (r *MetadataRetryer) QueueThought(id uuid.UUID) {
|
||||
return
|
||||
}
|
||||
r.logger.Info("background metadata complete",
|
||||
slog.String("thought_id", id.String()),
|
||||
slog.String("thought_id", idStr),
|
||||
slog.String("provider", r.metadata.PrimaryProvider()),
|
||||
slog.String("model", r.metadata.PrimaryModel()),
|
||||
slog.Bool("updated", updated),
|
||||
@@ -156,9 +157,9 @@ func (r *MetadataRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest,
|
||||
return nil, RetryMetadataOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
thoughts, err := r.store.ListThoughtsPendingMetadataRetry(ctx, limit, projectID, in.IncludeArchived, in.OlderThanDays)
|
||||
@@ -205,7 +206,7 @@ func (r *MetadataRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest,
|
||||
r.lock.Release(thought.ID)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
out.Failures = append(out.Failures, RetryMetadataFailure{ID: thought.ID.String(), Error: err.Error()})
|
||||
out.Failures = append(out.Failures, RetryMetadataFailure{ID: fmt.Sprint(thought.ID), Error: err.Error()})
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
@@ -228,8 +229,8 @@ func (r *MetadataRetryer) Handle(ctx context.Context, req *mcp.CallToolRequest,
|
||||
return nil, out, nil
|
||||
}
|
||||
|
||||
func (r *MetadataRetryer) retryOne(ctx context.Context, id uuid.UUID) (bool, error) {
|
||||
thought, err := r.store.GetThought(ctx, id)
|
||||
func (r *MetadataRetryer) retryOne(ctx context.Context, id int64) (bool, error) {
|
||||
thought, err := r.store.GetThoughtByID(ctx, id)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
@@ -5,7 +5,6 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/config"
|
||||
@@ -34,7 +33,7 @@ type CreatePlanInput struct {
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id; falls back to active session project"`
|
||||
Owner string `json:"owner,omitempty"`
|
||||
DueDate string `json:"due_date,omitempty" jsonschema:"RFC3339 timestamp"`
|
||||
SupersedesPlanID *uuid.UUID `json:"supersedes_plan_id,omitempty"`
|
||||
SupersedesPlanID *int64 `json:"supersedes_plan_id,omitempty"`
|
||||
Tags []string `json:"tags,omitempty"`
|
||||
}
|
||||
|
||||
@@ -43,7 +42,7 @@ type CreatePlanOutput struct {
|
||||
}
|
||||
|
||||
type GetPlanInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"plan id"`
|
||||
ID int64 `json:"id" jsonschema:"plan id"`
|
||||
}
|
||||
|
||||
type GetPlanOutput struct {
|
||||
@@ -51,7 +50,7 @@ type GetPlanOutput struct {
|
||||
}
|
||||
|
||||
type UpdatePlanInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"plan id"`
|
||||
ID int64 `json:"id" jsonschema:"plan id"`
|
||||
Title string `json:"title,omitempty"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Status string `json:"status,omitempty" jsonschema:"draft|active|blocked|completed|cancelled|superseded"`
|
||||
@@ -63,7 +62,7 @@ type UpdatePlanInput struct {
|
||||
ClearCompletedAt bool `json:"clear_completed_at,omitempty"`
|
||||
ReviewedBy *string `json:"reviewed_by,omitempty" jsonschema:"empty string clears the reviewer"`
|
||||
MarkReviewed bool `json:"mark_reviewed,omitempty" jsonschema:"set last_reviewed_at to now"`
|
||||
SupersedesPlanID *uuid.UUID `json:"supersedes_plan_id,omitempty"`
|
||||
SupersedesPlanID *int64 `json:"supersedes_plan_id,omitempty"`
|
||||
ClearSupersedesPlanID bool `json:"clear_supersedes_plan_id,omitempty"`
|
||||
Tags *[]string `json:"tags,omitempty" jsonschema:"replaces all tags when provided; pass [] to clear"`
|
||||
}
|
||||
@@ -73,7 +72,7 @@ type UpdatePlanOutput struct {
|
||||
}
|
||||
|
||||
type DeletePlanInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"plan id"`
|
||||
ID int64 `json:"id" jsonschema:"plan id"`
|
||||
}
|
||||
|
||||
type DeletePlanOutput struct {
|
||||
@@ -95,13 +94,13 @@ type ListPlansOutput struct {
|
||||
}
|
||||
|
||||
type PlanDependencyInput struct {
|
||||
PlanID uuid.UUID `json:"plan_id" jsonschema:"the plan that depends on another"`
|
||||
DependsOnPlanID uuid.UUID `json:"depends_on_plan_id" jsonschema:"the plan that must complete first"`
|
||||
PlanID int64 `json:"plan_id" jsonschema:"the plan that depends on another"`
|
||||
DependsOnPlanID int64 `json:"depends_on_plan_id" jsonschema:"the plan that must complete first"`
|
||||
}
|
||||
|
||||
type PlanRelatedInput struct {
|
||||
PlanAID uuid.UUID `json:"plan_a_id"`
|
||||
PlanBID uuid.UUID `json:"plan_b_id"`
|
||||
PlanAID int64 `json:"plan_a_id"`
|
||||
PlanBID int64 `json:"plan_b_id"`
|
||||
}
|
||||
|
||||
type PlanLinkOutput struct {
|
||||
@@ -109,12 +108,12 @@ type PlanLinkOutput struct {
|
||||
}
|
||||
|
||||
type PlanSkillInput struct {
|
||||
PlanID uuid.UUID `json:"plan_id"`
|
||||
SkillID uuid.UUID `json:"skill_id"`
|
||||
PlanID int64 `json:"plan_id"`
|
||||
SkillID int64 `json:"skill_id"`
|
||||
}
|
||||
|
||||
type ListPlanSkillsInput struct {
|
||||
PlanID uuid.UUID `json:"plan_id"`
|
||||
PlanID int64 `json:"plan_id"`
|
||||
}
|
||||
|
||||
type ListPlanSkillsOutput struct {
|
||||
@@ -122,12 +121,12 @@ type ListPlanSkillsOutput struct {
|
||||
}
|
||||
|
||||
type PlanGuardrailInput struct {
|
||||
PlanID uuid.UUID `json:"plan_id"`
|
||||
GuardrailID uuid.UUID `json:"guardrail_id"`
|
||||
PlanID int64 `json:"plan_id"`
|
||||
GuardrailID int64 `json:"guardrail_id"`
|
||||
}
|
||||
|
||||
type ListPlanGuardrailsInput struct {
|
||||
PlanID uuid.UUID `json:"plan_id"`
|
||||
PlanID int64 `json:"plan_id"`
|
||||
}
|
||||
|
||||
type ListPlanGuardrailsOutput struct {
|
||||
@@ -157,7 +156,7 @@ func (t *PlansTool) Create(ctx context.Context, req *mcp.CallToolRequest, in Cre
|
||||
Tags: normalizeStringSlice(in.Tags),
|
||||
}
|
||||
if project != nil {
|
||||
plan.ProjectID = &project.ID
|
||||
plan.ProjectID = &project.NumericID
|
||||
}
|
||||
if v := strings.TrimSpace(in.DueDate); v != "" {
|
||||
t, err := time.Parse(time.RFC3339, v)
|
||||
@@ -249,7 +248,7 @@ func (t *PlansTool) List(ctx context.Context, req *mcp.CallToolRequest, in ListP
|
||||
Query: strings.TrimSpace(in.Query),
|
||||
}
|
||||
if project != nil {
|
||||
filter.ProjectID = &project.ID
|
||||
filter.ProjectID = &project.NumericID
|
||||
}
|
||||
|
||||
plans, err := t.store.ListPlans(ctx, filter)
|
||||
|
||||
@@ -79,7 +79,7 @@ func (t *ProjectsTool) SetActive(ctx context.Context, req *mcp.CallToolRequest,
|
||||
return nil, SetActiveProjectOutput{}, err
|
||||
}
|
||||
t.sessions.Set(sid, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
return nil, SetActiveProjectOutput{Project: *project}, nil
|
||||
}
|
||||
|
||||
|
||||
@@ -5,7 +5,6 @@ import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/ai"
|
||||
@@ -49,9 +48,9 @@ func (t *RecallTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in Re
|
||||
|
||||
limit := normalizeLimit(in.Limit, t.search)
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
semantic, err := semanticSearch(ctx, t.store, t.embeddings, t.search, query, limit, t.search.DefaultThreshold, projectID, nil)
|
||||
@@ -66,7 +65,7 @@ func (t *RecallTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in Re
|
||||
items := make([]ContextItem, 0, limit*2)
|
||||
seen := map[string]struct{}{}
|
||||
for _, result := range semantic {
|
||||
key := result.ID.String()
|
||||
key := fmt.Sprint(result.ID)
|
||||
seen[key] = struct{}{}
|
||||
items = append(items, ContextItem{
|
||||
ID: key,
|
||||
@@ -77,7 +76,7 @@ func (t *RecallTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in Re
|
||||
})
|
||||
}
|
||||
for _, thought := range recent {
|
||||
key := thought.ID.String()
|
||||
key := fmt.Sprint(thought.ID)
|
||||
if _, ok := seen[key]; ok {
|
||||
continue
|
||||
}
|
||||
@@ -97,7 +96,7 @@ func (t *RecallTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in Re
|
||||
header := "Recalled context"
|
||||
if project != nil {
|
||||
header = fmt.Sprintf("Recalled context for %s", project.Name)
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
return nil, RecallOutput{
|
||||
|
||||
@@ -2,12 +2,12 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"reflect"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
"golang.org/x/sync/semaphore"
|
||||
|
||||
@@ -68,9 +68,9 @@ func (t *ReparseMetadataTool) Handle(ctx context.Context, req *mcp.CallToolReque
|
||||
return nil, ReparseMetadataOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
thoughts, err := t.store.ListThoughtsForMetadataReparse(ctx, limit, projectID, in.IncludeArchived, in.OlderThanDays)
|
||||
@@ -114,7 +114,7 @@ func (t *ReparseMetadataTool) Handle(ctx context.Context, req *mcp.CallToolReque
|
||||
mu.Lock()
|
||||
out.Normalized++
|
||||
mu.Unlock()
|
||||
t.logger.Warn("metadata reparse extract failed, using normalized existing metadata", slog.String("thought_id", thought.ID.String()), slog.String("error", extractErr.Error()))
|
||||
t.logger.Warn("metadata reparse extract failed, using normalized existing metadata", slog.String("thought_id", fmt.Sprint(thought.ID)), slog.String("error", extractErr.Error()))
|
||||
} else {
|
||||
normalizedTarget = metadata.MarkMetadataComplete(metadata.SanitizeExtracted(extracted), t.capture, attemptedAt)
|
||||
normalizedTarget.Attachments = thought.Metadata.Attachments
|
||||
@@ -130,11 +130,11 @@ func (t *ReparseMetadataTool) Handle(ctx context.Context, req *mcp.CallToolReque
|
||||
return
|
||||
}
|
||||
|
||||
if _, updateErr := t.store.UpdateThought(ctx, thought.ID, thought.Content, nil, "", normalizedTarget, thought.ProjectID); updateErr != nil {
|
||||
if _, updateErr := t.store.UpdateThought(ctx, thought.GUID, thought.Content, nil, "", normalizedTarget, thought.ProjectID); updateErr != nil {
|
||||
mu.Lock()
|
||||
out.Failures = append(out.Failures, ReparseMetadataFailure{ID: thought.ID.String(), Error: updateErr.Error()})
|
||||
out.Failures = append(out.Failures, ReparseMetadataFailure{ID: fmt.Sprint(thought.ID), Error: updateErr.Error()})
|
||||
mu.Unlock()
|
||||
t.logger.Warn("metadata reparse update failed", slog.String("thought_id", thought.ID.String()), slog.String("error", updateErr.Error()))
|
||||
t.logger.Warn("metadata reparse update failed", slog.String("thought_id", fmt.Sprint(thought.ID)), slog.String("error", updateErr.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@@ -23,7 +23,7 @@ func semanticSearch(
|
||||
query string,
|
||||
limit int,
|
||||
threshold float64,
|
||||
projectID *uuid.UUID,
|
||||
projectID *int64,
|
||||
excludeID *uuid.UUID,
|
||||
) ([]thoughttypes.SearchResult, error) {
|
||||
model := embeddings.PrimaryModel()
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/ai"
|
||||
@@ -50,10 +49,10 @@ func (t *SearchTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in Se
|
||||
return nil, SearchOutput{}, err
|
||||
}
|
||||
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
projectID = &project.NumericID
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
results, err := semanticSearch(ctx, t.store, t.embeddings, t.search, query, limit, threshold, projectID, nil)
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/session"
|
||||
@@ -59,7 +58,7 @@ func (t *SkillsTool) AddSkill(ctx context.Context, _ *mcp.CallToolRequest, in Ad
|
||||
// remove_skill
|
||||
|
||||
type RemoveSkillInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"skill id to remove"`
|
||||
ID int64 `json:"id" jsonschema:"skill id to remove"`
|
||||
}
|
||||
|
||||
type RemoveSkillOutput struct {
|
||||
@@ -147,7 +146,7 @@ func (t *SkillsTool) AddGuardrail(ctx context.Context, _ *mcp.CallToolRequest, i
|
||||
// remove_guardrail
|
||||
|
||||
type RemoveGuardrailInput struct {
|
||||
ID uuid.UUID `json:"id" jsonschema:"guardrail id to remove"`
|
||||
ID int64 `json:"id" jsonschema:"guardrail id to remove"`
|
||||
}
|
||||
|
||||
type RemoveGuardrailOutput struct {
|
||||
@@ -187,12 +186,12 @@ func (t *SkillsTool) ListGuardrails(ctx context.Context, _ *mcp.CallToolRequest,
|
||||
|
||||
type AddProjectSkillInput struct {
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id (uses active project if omitted)"`
|
||||
SkillID uuid.UUID `json:"skill_id" jsonschema:"skill id to link"`
|
||||
SkillID int64 `json:"skill_id" jsonschema:"skill id to link"`
|
||||
}
|
||||
|
||||
type AddProjectSkillOutput struct {
|
||||
ProjectID uuid.UUID `json:"project_id"`
|
||||
SkillID uuid.UUID `json:"skill_id"`
|
||||
ProjectID int64 `json:"project_id"`
|
||||
SkillID int64 `json:"skill_id"`
|
||||
}
|
||||
|
||||
func (t *SkillsTool) AddProjectSkill(ctx context.Context, req *mcp.CallToolRequest, in AddProjectSkillInput) (*mcp.CallToolResult, AddProjectSkillOutput, error) {
|
||||
@@ -200,17 +199,17 @@ func (t *SkillsTool) AddProjectSkill(ctx context.Context, req *mcp.CallToolReque
|
||||
if err != nil {
|
||||
return nil, AddProjectSkillOutput{}, err
|
||||
}
|
||||
if err := t.store.AddProjectSkill(ctx, project.ID, in.SkillID); err != nil {
|
||||
if err := t.store.AddProjectSkill(ctx, project.NumericID, in.SkillID); err != nil {
|
||||
return nil, AddProjectSkillOutput{}, err
|
||||
}
|
||||
return nil, AddProjectSkillOutput{ProjectID: project.ID, SkillID: in.SkillID}, nil
|
||||
return nil, AddProjectSkillOutput{ProjectID: project.NumericID, SkillID: in.SkillID}, nil
|
||||
}
|
||||
|
||||
// remove_project_skill
|
||||
|
||||
type RemoveProjectSkillInput struct {
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id (uses active project if omitted)"`
|
||||
SkillID uuid.UUID `json:"skill_id" jsonschema:"skill id to unlink"`
|
||||
SkillID int64 `json:"skill_id" jsonschema:"skill id to unlink"`
|
||||
}
|
||||
|
||||
type RemoveProjectSkillOutput struct {
|
||||
@@ -222,7 +221,7 @@ func (t *SkillsTool) RemoveProjectSkill(ctx context.Context, req *mcp.CallToolRe
|
||||
if err != nil {
|
||||
return nil, RemoveProjectSkillOutput{}, err
|
||||
}
|
||||
if err := t.store.RemoveProjectSkill(ctx, project.ID, in.SkillID); err != nil {
|
||||
if err := t.store.RemoveProjectSkill(ctx, project.NumericID, in.SkillID); err != nil {
|
||||
return nil, RemoveProjectSkillOutput{}, err
|
||||
}
|
||||
return nil, RemoveProjectSkillOutput{Removed: true}, nil
|
||||
@@ -235,7 +234,7 @@ type ListProjectSkillsInput struct {
|
||||
}
|
||||
|
||||
type ListProjectSkillsOutput struct {
|
||||
ProjectID uuid.UUID `json:"project_id"`
|
||||
ProjectID int64 `json:"project_id"`
|
||||
Skills []ext.AgentSkill `json:"skills"`
|
||||
}
|
||||
|
||||
@@ -244,26 +243,26 @@ func (t *SkillsTool) ListProjectSkills(ctx context.Context, req *mcp.CallToolReq
|
||||
if err != nil {
|
||||
return nil, ListProjectSkillsOutput{}, err
|
||||
}
|
||||
skills, err := t.store.ListProjectSkills(ctx, project.ID)
|
||||
skills, err := t.store.ListProjectSkills(ctx, project.NumericID)
|
||||
if err != nil {
|
||||
return nil, ListProjectSkillsOutput{}, err
|
||||
}
|
||||
if skills == nil {
|
||||
skills = []ext.AgentSkill{}
|
||||
}
|
||||
return nil, ListProjectSkillsOutput{ProjectID: project.ID, Skills: skills}, nil
|
||||
return nil, ListProjectSkillsOutput{ProjectID: project.NumericID, Skills: skills}, nil
|
||||
}
|
||||
|
||||
// add_project_guardrail
|
||||
|
||||
type AddProjectGuardrailInput struct {
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id (uses active project if omitted)"`
|
||||
GuardrailID uuid.UUID `json:"guardrail_id" jsonschema:"guardrail id to link"`
|
||||
GuardrailID int64 `json:"guardrail_id" jsonschema:"guardrail id to link"`
|
||||
}
|
||||
|
||||
type AddProjectGuardrailOutput struct {
|
||||
ProjectID uuid.UUID `json:"project_id"`
|
||||
GuardrailID uuid.UUID `json:"guardrail_id"`
|
||||
ProjectID int64 `json:"project_id"`
|
||||
GuardrailID int64 `json:"guardrail_id"`
|
||||
}
|
||||
|
||||
func (t *SkillsTool) AddProjectGuardrail(ctx context.Context, req *mcp.CallToolRequest, in AddProjectGuardrailInput) (*mcp.CallToolResult, AddProjectGuardrailOutput, error) {
|
||||
@@ -271,17 +270,17 @@ func (t *SkillsTool) AddProjectGuardrail(ctx context.Context, req *mcp.CallToolR
|
||||
if err != nil {
|
||||
return nil, AddProjectGuardrailOutput{}, err
|
||||
}
|
||||
if err := t.store.AddProjectGuardrail(ctx, project.ID, in.GuardrailID); err != nil {
|
||||
if err := t.store.AddProjectGuardrail(ctx, project.NumericID, in.GuardrailID); err != nil {
|
||||
return nil, AddProjectGuardrailOutput{}, err
|
||||
}
|
||||
return nil, AddProjectGuardrailOutput{ProjectID: project.ID, GuardrailID: in.GuardrailID}, nil
|
||||
return nil, AddProjectGuardrailOutput{ProjectID: project.NumericID, GuardrailID: in.GuardrailID}, nil
|
||||
}
|
||||
|
||||
// remove_project_guardrail
|
||||
|
||||
type RemoveProjectGuardrailInput struct {
|
||||
Project string `json:"project,omitempty" jsonschema:"project name or id (uses active project if omitted)"`
|
||||
GuardrailID uuid.UUID `json:"guardrail_id" jsonschema:"guardrail id to unlink"`
|
||||
GuardrailID int64 `json:"guardrail_id" jsonschema:"guardrail id to unlink"`
|
||||
}
|
||||
|
||||
type RemoveProjectGuardrailOutput struct {
|
||||
@@ -293,7 +292,7 @@ func (t *SkillsTool) RemoveProjectGuardrail(ctx context.Context, req *mcp.CallTo
|
||||
if err != nil {
|
||||
return nil, RemoveProjectGuardrailOutput{}, err
|
||||
}
|
||||
if err := t.store.RemoveProjectGuardrail(ctx, project.ID, in.GuardrailID); err != nil {
|
||||
if err := t.store.RemoveProjectGuardrail(ctx, project.NumericID, in.GuardrailID); err != nil {
|
||||
return nil, RemoveProjectGuardrailOutput{}, err
|
||||
}
|
||||
return nil, RemoveProjectGuardrailOutput{Removed: true}, nil
|
||||
@@ -306,7 +305,7 @@ type ListProjectGuardrailsInput struct {
|
||||
}
|
||||
|
||||
type ListProjectGuardrailsOutput struct {
|
||||
ProjectID uuid.UUID `json:"project_id"`
|
||||
ProjectID int64 `json:"project_id"`
|
||||
Guardrails []ext.AgentGuardrail `json:"guardrails"`
|
||||
}
|
||||
|
||||
@@ -315,12 +314,12 @@ func (t *SkillsTool) ListProjectGuardrails(ctx context.Context, req *mcp.CallToo
|
||||
if err != nil {
|
||||
return nil, ListProjectGuardrailsOutput{}, err
|
||||
}
|
||||
guardrails, err := t.store.ListProjectGuardrails(ctx, project.ID)
|
||||
guardrails, err := t.store.ListProjectGuardrails(ctx, project.NumericID)
|
||||
if err != nil {
|
||||
return nil, ListProjectGuardrailsOutput{}, err
|
||||
}
|
||||
if guardrails == nil {
|
||||
guardrails = []ext.AgentGuardrail{}
|
||||
}
|
||||
return nil, ListProjectGuardrailsOutput{ProjectID: project.ID, Guardrails: guardrails}, nil
|
||||
return nil, ListProjectGuardrailsOutput{ProjectID: project.NumericID, Guardrails: guardrails}, nil
|
||||
}
|
||||
|
||||
@@ -4,7 +4,6 @@ import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
"git.warky.dev/wdevs/amcs/internal/ai"
|
||||
@@ -49,9 +48,9 @@ func (t *SummarizeTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in
|
||||
count := 0
|
||||
|
||||
if query != "" {
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
results, err := semanticSearch(ctx, t.store, t.embeddings, t.search, query, limit, t.search.DefaultThreshold, projectID, nil)
|
||||
if err != nil {
|
||||
@@ -62,9 +61,9 @@ func (t *SummarizeTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in
|
||||
}
|
||||
count = len(results)
|
||||
} else {
|
||||
var projectID *uuid.UUID
|
||||
var projectID *int64
|
||||
if project != nil {
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
thoughts, err := t.store.RecentThoughts(ctx, projectID, limit, in.Days)
|
||||
if err != nil {
|
||||
@@ -83,7 +82,7 @@ func (t *SummarizeTool) Handle(ctx context.Context, req *mcp.CallToolRequest, in
|
||||
return nil, SummarizeOutput{}, err
|
||||
}
|
||||
if project != nil {
|
||||
_ = t.store.TouchProject(ctx, project.ID)
|
||||
_ = t.store.TouchProject(ctx, project.NumericID)
|
||||
}
|
||||
|
||||
return nil, SummarizeOutput{Summary: summary, Count: count}, nil
|
||||
|
||||
@@ -83,7 +83,7 @@ func (t *UpdateTool) Handle(ctx context.Context, _ *mcp.CallToolRequest, in Upda
|
||||
if err != nil {
|
||||
return nil, UpdateOutput{}, err
|
||||
}
|
||||
projectID = &project.ID
|
||||
projectID = &project.NumericID
|
||||
}
|
||||
|
||||
updated, err := t.store.UpdateThought(ctx, id, content, embedding, embeddingModel, mergedMetadata, projectID)
|
||||
|
||||
@@ -217,7 +217,8 @@ type ContactHistory struct {
|
||||
// Agent Skills & Guardrails
|
||||
|
||||
type AgentSkill struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Content string `json:"content"`
|
||||
@@ -227,7 +228,8 @@ type AgentSkill struct {
|
||||
}
|
||||
|
||||
type AgentGuardrail struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description,omitempty"`
|
||||
Content string `json:"content"`
|
||||
@@ -245,12 +247,13 @@ type ChatMessage struct {
|
||||
}
|
||||
|
||||
type ChatHistory struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
SessionID string `json:"session_id"`
|
||||
Title string `json:"title,omitempty"`
|
||||
Channel string `json:"channel,omitempty"`
|
||||
AgentID string `json:"agent_id,omitempty"`
|
||||
ProjectID *uuid.UUID `json:"project_id,omitempty"`
|
||||
ProjectID *int64 `json:"project_id,omitempty"`
|
||||
Messages []ChatMessage `json:"messages"`
|
||||
Summary string `json:"summary,omitempty"`
|
||||
Metadata map[string]any `json:"metadata"`
|
||||
|
||||
@@ -33,7 +33,8 @@ const (
|
||||
)
|
||||
|
||||
type Learning struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
Summary string `json:"summary"`
|
||||
Details string `json:"details"`
|
||||
Category string `json:"category"`
|
||||
@@ -44,13 +45,13 @@ type Learning struct {
|
||||
ActionRequired bool `json:"action_required"`
|
||||
SourceType string `json:"source_type,omitempty"`
|
||||
SourceRef string `json:"source_ref,omitempty"`
|
||||
ProjectID *uuid.UUID `json:"project_id,omitempty"`
|
||||
RelatedThoughtID *uuid.UUID `json:"related_thought_id,omitempty"`
|
||||
RelatedSkillID *uuid.UUID `json:"related_skill_id,omitempty"`
|
||||
ProjectID *int64 `json:"project_id,omitempty"`
|
||||
RelatedThoughtID *int64 `json:"related_thought_id,omitempty"`
|
||||
RelatedSkillID *int64 `json:"related_skill_id,omitempty"`
|
||||
ReviewedBy *string `json:"reviewed_by,omitempty"`
|
||||
ReviewedAt *time.Time `json:"reviewed_at,omitempty"`
|
||||
DuplicateOfLearningID *uuid.UUID `json:"duplicate_of_learning_id,omitempty"`
|
||||
SupersedesLearningID *uuid.UUID `json:"supersedes_learning_id,omitempty"`
|
||||
DuplicateOfLearningID *int64 `json:"duplicate_of_learning_id,omitempty"`
|
||||
SupersedesLearningID *int64 `json:"supersedes_learning_id,omitempty"`
|
||||
Tags []string `json:"tags"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
@@ -58,7 +59,7 @@ type Learning struct {
|
||||
|
||||
type LearningFilter struct {
|
||||
Limit int
|
||||
ProjectID *uuid.UUID
|
||||
ProjectID *int64
|
||||
Category string
|
||||
Area string
|
||||
Status string
|
||||
|
||||
@@ -27,18 +27,19 @@ const (
|
||||
)
|
||||
|
||||
type Plan struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
Title string `json:"title"`
|
||||
Description string `json:"description"`
|
||||
Status PlanStatus `json:"status"`
|
||||
Priority PlanPriority `json:"priority"`
|
||||
ProjectID *uuid.UUID `json:"project_id,omitempty"`
|
||||
ProjectID *int64 `json:"project_id,omitempty"`
|
||||
Owner string `json:"owner,omitempty"`
|
||||
DueDate *time.Time `json:"due_date,omitempty"`
|
||||
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
||||
ReviewedBy string `json:"reviewed_by,omitempty"`
|
||||
LastReviewedAt *time.Time `json:"last_reviewed_at,omitempty"`
|
||||
SupersedesPlanID *uuid.UUID `json:"supersedes_plan_id,omitempty"`
|
||||
SupersedesPlanID *int64 `json:"supersedes_plan_id,omitempty"`
|
||||
Tags []string `json:"tags"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
@@ -56,7 +57,7 @@ type PlanDetail struct {
|
||||
|
||||
type PlanFilter struct {
|
||||
Limit int
|
||||
ProjectID *uuid.UUID
|
||||
ProjectID *int64
|
||||
Status string
|
||||
Priority string
|
||||
Owner string
|
||||
@@ -77,7 +78,7 @@ type PlanUpdate struct {
|
||||
ClearCompletedAt bool
|
||||
ReviewedBy *string // "" to clear
|
||||
MarkReviewed bool // sets last_reviewed_at = now()
|
||||
SupersedesPlanID *uuid.UUID
|
||||
SupersedesPlanID *int64
|
||||
ClearSupersedesPlanID bool
|
||||
Tags *[]string // nil = no change; replace when non-nil
|
||||
}
|
||||
|
||||
@@ -9,11 +9,12 @@ import (
|
||||
type ThoughtPatch struct {
|
||||
Content *string `json:"content,omitempty"`
|
||||
Metadata ThoughtMetadata `json:"metadata,omitempty"`
|
||||
Project *uuid.UUID `json:"project_id,omitempty"`
|
||||
Project *int64 `json:"project_id,omitempty"`
|
||||
}
|
||||
|
||||
type Project struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
NumericID int64 `json:"-"`
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description,omitempty"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
|
||||
@@ -30,9 +30,10 @@ type ThoughtAttachment struct {
|
||||
}
|
||||
|
||||
type StoredFile struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ThoughtID *uuid.UUID `json:"thought_id,omitempty"`
|
||||
ProjectID *uuid.UUID `json:"project_id,omitempty"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
ThoughtID *int64 `json:"thought_id,omitempty"`
|
||||
ProjectID *int64 `json:"project_id,omitempty"`
|
||||
Name string `json:"name"`
|
||||
MediaType string `json:"media_type"`
|
||||
Kind string `json:"kind"`
|
||||
@@ -46,25 +47,26 @@ type StoredFile struct {
|
||||
|
||||
type StoredFileFilter struct {
|
||||
Limit int
|
||||
ThoughtID *uuid.UUID
|
||||
ProjectID *uuid.UUID
|
||||
ThoughtID *int64
|
||||
ProjectID *int64
|
||||
Kind string
|
||||
}
|
||||
|
||||
type Thought struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
GUID uuid.UUID `json:"guid"`
|
||||
Content string `json:"content"`
|
||||
Embedding []float32 `json:"embedding,omitempty"`
|
||||
EmbeddingStatus string `json:"embedding_status,omitempty"`
|
||||
Metadata ThoughtMetadata `json:"metadata"`
|
||||
ProjectID *uuid.UUID `json:"project_id,omitempty"`
|
||||
ProjectID *int64 `json:"project_id,omitempty"`
|
||||
ArchivedAt *time.Time `json:"archived_at,omitempty"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
type SearchResult struct {
|
||||
ID uuid.UUID `json:"id"`
|
||||
ID int64 `json:"id"`
|
||||
Content string `json:"content"`
|
||||
Metadata ThoughtMetadata `json:"metadata"`
|
||||
Similarity float64 `json:"similarity"`
|
||||
@@ -77,7 +79,7 @@ type ListFilter struct {
|
||||
Topic string
|
||||
Person string
|
||||
Days int
|
||||
ProjectID *uuid.UUID
|
||||
ProjectID *int64
|
||||
IncludeArchived bool
|
||||
}
|
||||
|
||||
|
||||
338
migrations/015_bigserial_guid_migration.sql
Normal file
338
migrations/015_bigserial_guid_migration.sql
Normal file
@@ -0,0 +1,338 @@
|
||||
-- Migration: Convert uuid PKs to bigserial, rename old uuid id to guid, convert FK columns to bigint
|
||||
-- Tables with uuid PK → bigserial: agent_skills, agent_guardrails, chat_histories, learnings, plans
|
||||
-- Tables with serial → bigserial: thought_links, plan_dependencies, plan_related_plans, plan_skills, plan_guardrails, project_skills, project_guardrails
|
||||
-- FK columns: all uuid FK columns converted to bigint referencing .id
|
||||
|
||||
BEGIN;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 1: Drop all FK constraints affected by this migration
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.thoughts DROP CONSTRAINT IF EXISTS thoughts_project_id_fkey;
|
||||
ALTER TABLE public.embeddings DROP CONSTRAINT IF EXISTS embeddings_thought_id_fkey;
|
||||
ALTER TABLE public.stored_files DROP CONSTRAINT IF EXISTS stored_files_thought_id_fkey;
|
||||
ALTER TABLE public.stored_files DROP CONSTRAINT IF EXISTS stored_files_project_id_fkey;
|
||||
ALTER TABLE public.project_skills DROP CONSTRAINT IF EXISTS project_skills_project_id_fkey;
|
||||
ALTER TABLE public.project_skills DROP CONSTRAINT IF EXISTS project_skills_skill_id_fkey;
|
||||
ALTER TABLE public.project_guardrails DROP CONSTRAINT IF EXISTS project_guardrails_project_id_fkey;
|
||||
ALTER TABLE public.project_guardrails DROP CONSTRAINT IF EXISTS project_guardrails_guardrail_id_fkey;
|
||||
ALTER TABLE public.chat_histories DROP CONSTRAINT IF EXISTS chat_histories_project_id_fkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_project_id_fkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_related_thought_id_fkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_related_skill_id_fkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_duplicate_of_learning_id_fkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_supersedes_learning_id_fkey;
|
||||
ALTER TABLE public.plans DROP CONSTRAINT IF EXISTS plans_project_id_fkey;
|
||||
ALTER TABLE public.plans DROP CONSTRAINT IF EXISTS plans_supersedes_plan_id_fkey;
|
||||
ALTER TABLE public.plan_dependencies DROP CONSTRAINT IF EXISTS plan_dependencies_plan_id_fkey;
|
||||
ALTER TABLE public.plan_dependencies DROP CONSTRAINT IF EXISTS plan_dependencies_depends_on_plan_id_fkey;
|
||||
ALTER TABLE public.plan_related_plans DROP CONSTRAINT IF EXISTS plan_related_plans_plan_a_id_fkey;
|
||||
ALTER TABLE public.plan_related_plans DROP CONSTRAINT IF EXISTS plan_related_plans_plan_b_id_fkey;
|
||||
ALTER TABLE public.plan_skills DROP CONSTRAINT IF EXISTS plan_skills_plan_id_fkey;
|
||||
ALTER TABLE public.plan_skills DROP CONSTRAINT IF EXISTS plan_skills_skill_id_fkey;
|
||||
ALTER TABLE public.plan_guardrails DROP CONSTRAINT IF EXISTS plan_guardrails_plan_id_fkey;
|
||||
ALTER TABLE public.plan_guardrails DROP CONSTRAINT IF EXISTS plan_guardrails_guardrail_id_fkey;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 2: Drop PK constraints on uuid-pk tables
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.agent_skills DROP CONSTRAINT IF EXISTS agent_skills_pkey;
|
||||
ALTER TABLE public.agent_guardrails DROP CONSTRAINT IF EXISTS agent_guardrails_pkey;
|
||||
ALTER TABLE public.chat_histories DROP CONSTRAINT IF EXISTS chat_histories_pkey;
|
||||
ALTER TABLE public.learnings DROP CONSTRAINT IF EXISTS learnings_pkey;
|
||||
ALTER TABLE public.plans DROP CONSTRAINT IF EXISTS plans_pkey;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 3: Rename uuid id → guid on formerly uuid-pk tables
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.agent_skills RENAME COLUMN id TO guid;
|
||||
ALTER TABLE public.agent_guardrails RENAME COLUMN id TO guid;
|
||||
ALTER TABLE public.chat_histories RENAME COLUMN id TO guid;
|
||||
ALTER TABLE public.learnings RENAME COLUMN id TO guid;
|
||||
ALTER TABLE public.plans RENAME COLUMN id TO guid;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 4: Add bigserial id columns to formerly uuid-pk tables
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.agent_skills ADD COLUMN id bigserial;
|
||||
ALTER TABLE public.agent_guardrails ADD COLUMN id bigserial;
|
||||
ALTER TABLE public.chat_histories ADD COLUMN id bigserial;
|
||||
ALTER TABLE public.learnings ADD COLUMN id bigserial;
|
||||
ALTER TABLE public.plans ADD COLUMN id bigserial;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 5: Add new bigint FK columns (prefixed _new) where uuid FKs exist
|
||||
-- ============================================================
|
||||
|
||||
-- thoughts.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.thoughts ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.thoughts t SET project_id_new = p.id FROM public.projects p WHERE p.guid = t.project_id;
|
||||
|
||||
-- embeddings.thought_id (uuid → bigint via thoughts.guid)
|
||||
ALTER TABLE public.embeddings ADD COLUMN thought_id_new bigint;
|
||||
UPDATE public.embeddings e SET thought_id_new = t.id FROM public.thoughts t WHERE t.guid = e.thought_id;
|
||||
|
||||
-- stored_files.thought_id (uuid → bigint via thoughts.guid)
|
||||
ALTER TABLE public.stored_files ADD COLUMN thought_id_new bigint;
|
||||
UPDATE public.stored_files f SET thought_id_new = t.id FROM public.thoughts t WHERE t.guid = f.thought_id;
|
||||
|
||||
-- stored_files.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.stored_files ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.stored_files f SET project_id_new = p.id FROM public.projects p WHERE p.guid = f.project_id;
|
||||
|
||||
-- project_skills.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.project_skills ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.project_skills ps SET project_id_new = p.id FROM public.projects p WHERE p.guid = ps.project_id;
|
||||
|
||||
-- project_skills.skill_id (uuid → bigint via agent_skills.guid)
|
||||
ALTER TABLE public.project_skills ADD COLUMN skill_id_new bigint;
|
||||
UPDATE public.project_skills ps SET skill_id_new = s.id FROM public.agent_skills s WHERE s.guid = ps.skill_id;
|
||||
|
||||
-- project_guardrails.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.project_guardrails ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.project_guardrails pg SET project_id_new = p.id FROM public.projects p WHERE p.guid = pg.project_id;
|
||||
|
||||
-- project_guardrails.guardrail_id (uuid → bigint via agent_guardrails.guid)
|
||||
ALTER TABLE public.project_guardrails ADD COLUMN guardrail_id_new bigint;
|
||||
UPDATE public.project_guardrails pg SET guardrail_id_new = g.id FROM public.agent_guardrails g WHERE g.guid = pg.guardrail_id;
|
||||
|
||||
-- chat_histories.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.chat_histories ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.chat_histories ch SET project_id_new = p.id FROM public.projects p WHERE p.guid = ch.project_id;
|
||||
|
||||
-- learnings.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.learnings ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.learnings l SET project_id_new = p.id FROM public.projects p WHERE p.guid = l.project_id;
|
||||
|
||||
-- learnings.related_thought_id (uuid → bigint via thoughts.guid)
|
||||
ALTER TABLE public.learnings ADD COLUMN related_thought_id_new bigint;
|
||||
UPDATE public.learnings l SET related_thought_id_new = t.id FROM public.thoughts t WHERE t.guid = l.related_thought_id;
|
||||
|
||||
-- learnings.related_skill_id (uuid → bigint via agent_skills.guid)
|
||||
ALTER TABLE public.learnings ADD COLUMN related_skill_id_new bigint;
|
||||
UPDATE public.learnings l SET related_skill_id_new = s.id FROM public.agent_skills s WHERE s.guid = l.related_skill_id;
|
||||
|
||||
-- learnings.duplicate_of_learning_id (uuid → bigint via learnings.guid)
|
||||
ALTER TABLE public.learnings ADD COLUMN duplicate_of_learning_id_new bigint;
|
||||
UPDATE public.learnings l SET duplicate_of_learning_id_new = l2.id FROM public.learnings l2 WHERE l2.guid = l.duplicate_of_learning_id;
|
||||
|
||||
-- learnings.supersedes_learning_id (uuid → bigint via learnings.guid)
|
||||
ALTER TABLE public.learnings ADD COLUMN supersedes_learning_id_new bigint;
|
||||
UPDATE public.learnings l SET supersedes_learning_id_new = l2.id FROM public.learnings l2 WHERE l2.guid = l.supersedes_learning_id;
|
||||
|
||||
-- plans.project_id (uuid → bigint via projects.guid)
|
||||
ALTER TABLE public.plans ADD COLUMN project_id_new bigint;
|
||||
UPDATE public.plans pl SET project_id_new = p.id FROM public.projects p WHERE p.guid = pl.project_id;
|
||||
|
||||
-- plans.supersedes_plan_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plans ADD COLUMN supersedes_plan_id_new bigint;
|
||||
UPDATE public.plans pl SET supersedes_plan_id_new = pl2.id FROM public.plans pl2 WHERE pl2.guid = pl.supersedes_plan_id;
|
||||
|
||||
-- plan_dependencies.plan_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_dependencies ADD COLUMN plan_id_new bigint;
|
||||
UPDATE public.plan_dependencies pd SET plan_id_new = pl.id FROM public.plans pl WHERE pl.guid = pd.plan_id;
|
||||
|
||||
-- plan_dependencies.depends_on_plan_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_dependencies ADD COLUMN depends_on_plan_id_new bigint;
|
||||
UPDATE public.plan_dependencies pd SET depends_on_plan_id_new = pl.id FROM public.plans pl WHERE pl.guid = pd.depends_on_plan_id;
|
||||
|
||||
-- plan_related_plans.plan_a_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_related_plans ADD COLUMN plan_a_id_new bigint;
|
||||
UPDATE public.plan_related_plans pr SET plan_a_id_new = pl.id FROM public.plans pl WHERE pl.guid = pr.plan_a_id;
|
||||
|
||||
-- plan_related_plans.plan_b_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_related_plans ADD COLUMN plan_b_id_new bigint;
|
||||
UPDATE public.plan_related_plans pr SET plan_b_id_new = pl.id FROM public.plans pl WHERE pl.guid = pr.plan_b_id;
|
||||
|
||||
-- plan_skills.plan_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_skills ADD COLUMN plan_id_new bigint;
|
||||
UPDATE public.plan_skills ps SET plan_id_new = pl.id FROM public.plans pl WHERE pl.guid = ps.plan_id;
|
||||
|
||||
-- plan_skills.skill_id (uuid → bigint via agent_skills.guid)
|
||||
ALTER TABLE public.plan_skills ADD COLUMN skill_id_new bigint;
|
||||
UPDATE public.plan_skills ps SET skill_id_new = s.id FROM public.agent_skills s WHERE s.guid = ps.skill_id;
|
||||
|
||||
-- plan_guardrails.plan_id (uuid → bigint via plans.guid)
|
||||
ALTER TABLE public.plan_guardrails ADD COLUMN plan_id_new bigint;
|
||||
UPDATE public.plan_guardrails pg SET plan_id_new = pl.id FROM public.plans pl WHERE pl.guid = pg.plan_id;
|
||||
|
||||
-- plan_guardrails.guardrail_id (uuid → bigint via agent_guardrails.guid)
|
||||
ALTER TABLE public.plan_guardrails ADD COLUMN guardrail_id_new bigint;
|
||||
UPDATE public.plan_guardrails pg SET guardrail_id_new = g.id FROM public.agent_guardrails g WHERE g.guid = pg.guardrail_id;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 6: Drop old uuid FK columns
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.thoughts DROP COLUMN project_id;
|
||||
ALTER TABLE public.embeddings DROP COLUMN thought_id;
|
||||
ALTER TABLE public.stored_files DROP COLUMN thought_id;
|
||||
ALTER TABLE public.stored_files DROP COLUMN project_id;
|
||||
ALTER TABLE public.project_skills DROP COLUMN project_id;
|
||||
ALTER TABLE public.project_skills DROP COLUMN skill_id;
|
||||
ALTER TABLE public.project_guardrails DROP COLUMN project_id;
|
||||
ALTER TABLE public.project_guardrails DROP COLUMN guardrail_id;
|
||||
ALTER TABLE public.chat_histories DROP COLUMN project_id;
|
||||
ALTER TABLE public.learnings DROP COLUMN project_id;
|
||||
ALTER TABLE public.learnings DROP COLUMN related_thought_id;
|
||||
ALTER TABLE public.learnings DROP COLUMN related_skill_id;
|
||||
ALTER TABLE public.learnings DROP COLUMN duplicate_of_learning_id;
|
||||
ALTER TABLE public.learnings DROP COLUMN supersedes_learning_id;
|
||||
ALTER TABLE public.plans DROP COLUMN project_id;
|
||||
ALTER TABLE public.plans DROP COLUMN supersedes_plan_id;
|
||||
ALTER TABLE public.plan_dependencies DROP COLUMN plan_id;
|
||||
ALTER TABLE public.plan_dependencies DROP COLUMN depends_on_plan_id;
|
||||
ALTER TABLE public.plan_related_plans DROP COLUMN plan_a_id;
|
||||
ALTER TABLE public.plan_related_plans DROP COLUMN plan_b_id;
|
||||
ALTER TABLE public.plan_skills DROP COLUMN plan_id;
|
||||
ALTER TABLE public.plan_skills DROP COLUMN skill_id;
|
||||
ALTER TABLE public.plan_guardrails DROP COLUMN plan_id;
|
||||
ALTER TABLE public.plan_guardrails DROP COLUMN guardrail_id;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 7: Rename _new columns to final names
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.thoughts RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.embeddings RENAME COLUMN thought_id_new TO thought_id;
|
||||
ALTER TABLE public.stored_files RENAME COLUMN thought_id_new TO thought_id;
|
||||
ALTER TABLE public.stored_files RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.project_skills RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.project_skills RENAME COLUMN skill_id_new TO skill_id;
|
||||
ALTER TABLE public.project_guardrails RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.project_guardrails RENAME COLUMN guardrail_id_new TO guardrail_id;
|
||||
ALTER TABLE public.chat_histories RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.learnings RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.learnings RENAME COLUMN related_thought_id_new TO related_thought_id;
|
||||
ALTER TABLE public.learnings RENAME COLUMN related_skill_id_new TO related_skill_id;
|
||||
ALTER TABLE public.learnings RENAME COLUMN duplicate_of_learning_id_new TO duplicate_of_learning_id;
|
||||
ALTER TABLE public.learnings RENAME COLUMN supersedes_learning_id_new TO supersedes_learning_id;
|
||||
ALTER TABLE public.plans RENAME COLUMN project_id_new TO project_id;
|
||||
ALTER TABLE public.plans RENAME COLUMN supersedes_plan_id_new TO supersedes_plan_id;
|
||||
ALTER TABLE public.plan_dependencies RENAME COLUMN plan_id_new TO plan_id;
|
||||
ALTER TABLE public.plan_dependencies RENAME COLUMN depends_on_plan_id_new TO depends_on_plan_id;
|
||||
ALTER TABLE public.plan_related_plans RENAME COLUMN plan_a_id_new TO plan_a_id;
|
||||
ALTER TABLE public.plan_related_plans RENAME COLUMN plan_b_id_new TO plan_b_id;
|
||||
ALTER TABLE public.plan_skills RENAME COLUMN plan_id_new TO plan_id;
|
||||
ALTER TABLE public.plan_skills RENAME COLUMN skill_id_new TO skill_id;
|
||||
ALTER TABLE public.plan_guardrails RENAME COLUMN plan_id_new TO plan_id;
|
||||
ALTER TABLE public.plan_guardrails RENAME COLUMN guardrail_id_new TO guardrail_id;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 8: Upgrade serial → bigserial (alter sequence type)
|
||||
-- ============================================================
|
||||
|
||||
-- thought_links
|
||||
ALTER SEQUENCE IF EXISTS thought_links_id_seq AS bigint;
|
||||
ALTER TABLE public.thought_links ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- plan_dependencies
|
||||
ALTER SEQUENCE IF EXISTS plan_dependencies_id_seq AS bigint;
|
||||
ALTER TABLE public.plan_dependencies ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- plan_related_plans
|
||||
ALTER SEQUENCE IF EXISTS plan_related_plans_id_seq AS bigint;
|
||||
ALTER TABLE public.plan_related_plans ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- plan_skills
|
||||
ALTER SEQUENCE IF EXISTS plan_skills_id_seq AS bigint;
|
||||
ALTER TABLE public.plan_skills ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- plan_guardrails
|
||||
ALTER SEQUENCE IF EXISTS plan_guardrails_id_seq AS bigint;
|
||||
ALTER TABLE public.plan_guardrails ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- project_skills
|
||||
ALTER SEQUENCE IF EXISTS project_skills_id_seq AS bigint;
|
||||
ALTER TABLE public.project_skills ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- project_guardrails
|
||||
ALTER SEQUENCE IF EXISTS project_guardrails_id_seq AS bigint;
|
||||
ALTER TABLE public.project_guardrails ALTER COLUMN id TYPE bigint;
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 9: Add PK constraints to formerly uuid-pk tables
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.agent_skills ADD CONSTRAINT agent_skills_pkey PRIMARY KEY (id);
|
||||
ALTER TABLE public.agent_guardrails ADD CONSTRAINT agent_guardrails_pkey PRIMARY KEY (id);
|
||||
ALTER TABLE public.chat_histories ADD CONSTRAINT chat_histories_pkey PRIMARY KEY (id);
|
||||
ALTER TABLE public.learnings ADD CONSTRAINT learnings_pkey PRIMARY KEY (id);
|
||||
ALTER TABLE public.plans ADD CONSTRAINT plans_pkey PRIMARY KEY (id);
|
||||
|
||||
-- Add unique constraint on guid columns
|
||||
ALTER TABLE public.agent_skills ADD CONSTRAINT agent_skills_guid_key UNIQUE (guid);
|
||||
ALTER TABLE public.agent_guardrails ADD CONSTRAINT agent_guardrails_guid_key UNIQUE (guid);
|
||||
ALTER TABLE public.chat_histories ADD CONSTRAINT chat_histories_guid_key UNIQUE (guid);
|
||||
ALTER TABLE public.learnings ADD CONSTRAINT learnings_guid_key UNIQUE (guid);
|
||||
ALTER TABLE public.plans ADD CONSTRAINT plans_guid_key UNIQUE (guid);
|
||||
|
||||
-- ============================================================
|
||||
-- STEP 10: Re-add FK constraints
|
||||
-- ============================================================
|
||||
|
||||
ALTER TABLE public.thoughts
|
||||
ADD CONSTRAINT thoughts_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id);
|
||||
|
||||
ALTER TABLE public.embeddings
|
||||
ADD CONSTRAINT embeddings_thought_id_fkey FOREIGN KEY (thought_id) REFERENCES public.thoughts(id);
|
||||
|
||||
ALTER TABLE public.stored_files
|
||||
ADD CONSTRAINT stored_files_thought_id_fkey FOREIGN KEY (thought_id) REFERENCES public.thoughts(id);
|
||||
ALTER TABLE public.stored_files
|
||||
ADD CONSTRAINT stored_files_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id);
|
||||
|
||||
ALTER TABLE public.project_skills
|
||||
ADD CONSTRAINT project_skills_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.project_skills
|
||||
ADD CONSTRAINT project_skills_skill_id_fkey FOREIGN KEY (skill_id) REFERENCES public.agent_skills(id) ON DELETE CASCADE;
|
||||
|
||||
ALTER TABLE public.project_guardrails
|
||||
ADD CONSTRAINT project_guardrails_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.project_guardrails
|
||||
ADD CONSTRAINT project_guardrails_guardrail_id_fkey FOREIGN KEY (guardrail_id) REFERENCES public.agent_guardrails(id) ON DELETE CASCADE;
|
||||
|
||||
ALTER TABLE public.chat_histories
|
||||
ADD CONSTRAINT chat_histories_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id) ON DELETE SET NULL;
|
||||
|
||||
ALTER TABLE public.learnings
|
||||
ADD CONSTRAINT learnings_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id) ON DELETE SET NULL;
|
||||
ALTER TABLE public.learnings
|
||||
ADD CONSTRAINT learnings_related_thought_id_fkey FOREIGN KEY (related_thought_id) REFERENCES public.thoughts(id) ON DELETE SET NULL;
|
||||
ALTER TABLE public.learnings
|
||||
ADD CONSTRAINT learnings_related_skill_id_fkey FOREIGN KEY (related_skill_id) REFERENCES public.agent_skills(id) ON DELETE SET NULL;
|
||||
ALTER TABLE public.learnings
|
||||
ADD CONSTRAINT learnings_duplicate_of_learning_id_fkey FOREIGN KEY (duplicate_of_learning_id) REFERENCES public.learnings(id) ON DELETE SET NULL;
|
||||
ALTER TABLE public.learnings
|
||||
ADD CONSTRAINT learnings_supersedes_learning_id_fkey FOREIGN KEY (supersedes_learning_id) REFERENCES public.learnings(id) ON DELETE SET NULL;
|
||||
|
||||
ALTER TABLE public.plans
|
||||
ADD CONSTRAINT plans_project_id_fkey FOREIGN KEY (project_id) REFERENCES public.projects(id) ON DELETE SET NULL;
|
||||
ALTER TABLE public.plans
|
||||
ADD CONSTRAINT plans_supersedes_plan_id_fkey FOREIGN KEY (supersedes_plan_id) REFERENCES public.plans(id) ON DELETE SET NULL;
|
||||
|
||||
ALTER TABLE public.plan_dependencies
|
||||
ADD CONSTRAINT plan_dependencies_plan_id_fkey FOREIGN KEY (plan_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.plan_dependencies
|
||||
ADD CONSTRAINT plan_dependencies_depends_on_plan_id_fkey FOREIGN KEY (depends_on_plan_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
|
||||
ALTER TABLE public.plan_related_plans
|
||||
ADD CONSTRAINT plan_related_plans_plan_a_id_fkey FOREIGN KEY (plan_a_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.plan_related_plans
|
||||
ADD CONSTRAINT plan_related_plans_plan_b_id_fkey FOREIGN KEY (plan_b_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
|
||||
ALTER TABLE public.plan_skills
|
||||
ADD CONSTRAINT plan_skills_plan_id_fkey FOREIGN KEY (plan_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.plan_skills
|
||||
ADD CONSTRAINT plan_skills_skill_id_fkey FOREIGN KEY (skill_id) REFERENCES public.agent_skills(id) ON DELETE CASCADE;
|
||||
|
||||
ALTER TABLE public.plan_guardrails
|
||||
ADD CONSTRAINT plan_guardrails_plan_id_fkey FOREIGN KEY (plan_id) REFERENCES public.plans(id) ON DELETE CASCADE;
|
||||
ALTER TABLE public.plan_guardrails
|
||||
ADD CONSTRAINT plan_guardrails_guardrail_id_fkey FOREIGN KEY (guardrail_id) REFERENCES public.agent_guardrails(id) ON DELETE CASCADE;
|
||||
|
||||
COMMIT;
|
||||
File diff suppressed because it is too large
Load Diff
@@ -5,7 +5,7 @@ Table thoughts {
|
||||
metadata jsonb [default: `'{}'::jsonb`]
|
||||
created_at timestamptz [default: `now()`]
|
||||
updated_at timestamptz [default: `now()`]
|
||||
project_id uuid [ref: > projects.guid]
|
||||
project_id bigint [ref: > projects.id]
|
||||
archived_at timestamptz
|
||||
}
|
||||
|
||||
@@ -19,7 +19,7 @@ Table projects {
|
||||
}
|
||||
|
||||
Table thought_links {
|
||||
id serial [pk]
|
||||
id bigserial [pk]
|
||||
from_id bigint [not null, ref: > thoughts.id]
|
||||
to_id bigint [not null, ref: > thoughts.id]
|
||||
relation text [not null]
|
||||
@@ -35,7 +35,7 @@ Table thought_links {
|
||||
Table embeddings {
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
thought_id uuid [not null, ref: > thoughts.guid]
|
||||
thought_id bigint [not null, ref: > thoughts.id]
|
||||
model text [not null]
|
||||
dim int [not null]
|
||||
embedding vector [not null]
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
Table stored_files {
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
thought_id uuid [ref: > thoughts.guid]
|
||||
project_id uuid [ref: > projects.guid]
|
||||
thought_id bigint [ref: > thoughts.id]
|
||||
project_id bigint [ref: > projects.id]
|
||||
name text [not null]
|
||||
media_type text [not null]
|
||||
kind text [not null, default: 'file']
|
||||
@@ -21,5 +21,5 @@ Table stored_files {
|
||||
}
|
||||
|
||||
// Cross-file refs (for relspecgo merge)
|
||||
Ref: stored_files.thought_id > thoughts.guid [delete: set null]
|
||||
Ref: stored_files.project_id > projects.guid [delete: set null]
|
||||
Ref: stored_files.thought_id > thoughts.id [delete: set null]
|
||||
Ref: stored_files.project_id > projects.id [delete: set null]
|
||||
|
||||
@@ -1,10 +1,11 @@
|
||||
Table chat_histories {
|
||||
id uuid [pk, default: `gen_random_uuid()`]
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
session_id text [not null]
|
||||
title text
|
||||
channel text
|
||||
agent_id text
|
||||
project_id uuid [ref: > projects.guid]
|
||||
project_id bigint [ref: > projects.id]
|
||||
messages jsonb [not null, default: `'[]'`]
|
||||
summary text
|
||||
metadata jsonb [not null, default: `'{}'`]
|
||||
@@ -29,10 +30,11 @@ Table tool_annotations {
|
||||
}
|
||||
|
||||
// Cross-file refs (for relspecgo merge)
|
||||
Ref: chat_histories.project_id > projects.guid [delete: set null]
|
||||
Ref: chat_histories.project_id > projects.id [delete: set null]
|
||||
|
||||
Table learnings {
|
||||
id uuid [pk, default: `gen_random_uuid()`]
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
summary text [not null]
|
||||
details text [not null, default: '']
|
||||
category text [not null, default: 'insight']
|
||||
@@ -43,13 +45,13 @@ Table learnings {
|
||||
action_required boolean [not null, default: false]
|
||||
source_type text
|
||||
source_ref text
|
||||
project_id uuid [ref: > projects.guid]
|
||||
related_thought_id uuid [ref: > thoughts.guid]
|
||||
related_skill_id uuid [ref: > agent_skills.id]
|
||||
project_id bigint [ref: > projects.id]
|
||||
related_thought_id bigint [ref: > thoughts.id]
|
||||
related_skill_id bigint [ref: > agent_skills.id]
|
||||
reviewed_by text
|
||||
reviewed_at timestamptz
|
||||
duplicate_of_learning_id uuid [ref: > learnings.id]
|
||||
supersedes_learning_id uuid [ref: > learnings.id]
|
||||
duplicate_of_learning_id bigint [ref: > learnings.id]
|
||||
supersedes_learning_id bigint [ref: > learnings.id]
|
||||
tags "text[]" [not null, default: `'{}'`]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
updated_at timestamptz [not null, default: `now()`]
|
||||
@@ -68,8 +70,8 @@ Table learnings {
|
||||
}
|
||||
|
||||
// Cross-file refs (for relspecgo merge)
|
||||
Ref: learnings.project_id > projects.guid [delete: set null]
|
||||
Ref: learnings.related_thought_id > thoughts.guid [delete: set null]
|
||||
Ref: learnings.project_id > projects.id [delete: set null]
|
||||
Ref: learnings.related_thought_id > thoughts.id [delete: set null]
|
||||
Ref: learnings.related_skill_id > agent_skills.id [delete: set null]
|
||||
Ref: learnings.duplicate_of_learning_id > learnings.id [delete: set null]
|
||||
Ref: learnings.supersedes_learning_id > learnings.id [delete: set null]
|
||||
|
||||
@@ -1,16 +1,17 @@
|
||||
Table plans {
|
||||
id uuid [pk, default: `gen_random_uuid()`]
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
title text [not null]
|
||||
description text [not null, default: '']
|
||||
status text [not null, default: 'draft'] // draft, active, blocked, completed, cancelled, superseded
|
||||
priority text [not null, default: 'medium'] // low, medium, high, critical
|
||||
project_id uuid [ref: > projects.guid]
|
||||
project_id bigint [ref: > projects.id]
|
||||
owner text
|
||||
due_date timestamptz
|
||||
completed_at timestamptz
|
||||
reviewed_by text
|
||||
last_reviewed_at timestamptz
|
||||
supersedes_plan_id uuid [ref: > plans.id]
|
||||
supersedes_plan_id bigint [ref: > plans.id]
|
||||
tags "text[]" [not null, default: `'{}'`]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
updated_at timestamptz [not null, default: `now()`]
|
||||
@@ -29,9 +30,9 @@ Table plans {
|
||||
|
||||
// Directional: plan_id cannot proceed until depends_on_plan_id is complete
|
||||
Table plan_dependencies {
|
||||
id serial [pk]
|
||||
plan_id uuid [not null, ref: > plans.id]
|
||||
depends_on_plan_id uuid [not null, ref: > plans.id]
|
||||
id bigserial [pk]
|
||||
plan_id bigint [not null, ref: > plans.id]
|
||||
depends_on_plan_id bigint [not null, ref: > plans.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -43,9 +44,9 @@ Table plan_dependencies {
|
||||
|
||||
// Bidirectional: store with plan_a_id < plan_b_id to avoid duplicates
|
||||
Table plan_related_plans {
|
||||
id serial [pk]
|
||||
plan_a_id uuid [not null, ref: > plans.id]
|
||||
plan_b_id uuid [not null, ref: > plans.id]
|
||||
id bigserial [pk]
|
||||
plan_a_id bigint [not null, ref: > plans.id]
|
||||
plan_b_id bigint [not null, ref: > plans.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -56,9 +57,9 @@ Table plan_related_plans {
|
||||
}
|
||||
|
||||
Table plan_skills {
|
||||
id serial [pk]
|
||||
plan_id uuid [not null, ref: > plans.id]
|
||||
skill_id uuid [not null, ref: > agent_skills.id]
|
||||
id bigserial [pk]
|
||||
plan_id bigint [not null, ref: > plans.id]
|
||||
skill_id bigint [not null, ref: > agent_skills.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -68,9 +69,9 @@ Table plan_skills {
|
||||
}
|
||||
|
||||
Table plan_guardrails {
|
||||
id serial [pk]
|
||||
plan_id uuid [not null, ref: > plans.id]
|
||||
guardrail_id uuid [not null, ref: > agent_guardrails.id]
|
||||
id bigserial [pk]
|
||||
plan_id bigint [not null, ref: > plans.id]
|
||||
guardrail_id bigint [not null, ref: > agent_guardrails.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -80,7 +81,7 @@ Table plan_guardrails {
|
||||
}
|
||||
|
||||
// Cross-file refs (for relspecgo merge)
|
||||
Ref: plans.project_id > projects.guid [delete: set null]
|
||||
Ref: plans.project_id > projects.id [delete: set null]
|
||||
Ref: plans.supersedes_plan_id > plans.id [delete: set null]
|
||||
Ref: plan_dependencies.plan_id > plans.id [delete: cascade]
|
||||
Ref: plan_dependencies.depends_on_plan_id > plans.id [delete: cascade]
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
Table agent_skills {
|
||||
id uuid [pk, default: `gen_random_uuid()`]
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
name text [unique, not null]
|
||||
description text [not null, default: '']
|
||||
content text [not null]
|
||||
@@ -9,7 +10,8 @@ Table agent_skills {
|
||||
}
|
||||
|
||||
Table agent_guardrails {
|
||||
id uuid [pk, default: `gen_random_uuid()`]
|
||||
id bigserial [pk]
|
||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||
name text [unique, not null]
|
||||
description text [not null, default: '']
|
||||
content text [not null]
|
||||
@@ -20,9 +22,9 @@ Table agent_guardrails {
|
||||
}
|
||||
|
||||
Table project_skills {
|
||||
id serial [pk]
|
||||
project_id uuid [not null, ref: > projects.guid]
|
||||
skill_id uuid [not null, ref: > agent_skills.id]
|
||||
id bigserial [pk]
|
||||
project_id bigint [not null, ref: > projects.id]
|
||||
skill_id bigint [not null, ref: > agent_skills.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -32,9 +34,9 @@ Table project_skills {
|
||||
}
|
||||
|
||||
Table project_guardrails {
|
||||
id serial [pk]
|
||||
project_id uuid [not null, ref: > projects.guid]
|
||||
guardrail_id uuid [not null, ref: > agent_guardrails.id]
|
||||
id bigserial [pk]
|
||||
project_id bigint [not null, ref: > projects.id]
|
||||
guardrail_id bigint [not null, ref: > agent_guardrails.id]
|
||||
created_at timestamptz [not null, default: `now()`]
|
||||
|
||||
indexes {
|
||||
@@ -44,5 +46,7 @@ Table project_guardrails {
|
||||
}
|
||||
|
||||
// Cross-file refs (for relspecgo merge)
|
||||
Ref: project_skills.project_id > projects.guid [delete: cascade]
|
||||
Ref: project_guardrails.project_id > projects.guid [delete: cascade]
|
||||
Ref: project_skills.project_id > projects.id [delete: cascade]
|
||||
Ref: project_skills.skill_id > agent_skills.id [delete: cascade]
|
||||
Ref: project_guardrails.project_id > projects.id [delete: cascade]
|
||||
Ref: project_guardrails.guardrail_id > agent_guardrails.id [delete: cascade]
|
||||
|
||||
Reference in New Issue
Block a user