diff --git a/packages/cli-go/internal/cmd/ingest.go b/packages/cli-go/internal/cmd/ingest.go index 73a342b..c1ec3b9 100644 --- a/packages/cli-go/internal/cmd/ingest.go +++ b/packages/cli-go/internal/cmd/ingest.go @@ -43,10 +43,10 @@ func doIngest(content, sourceType string) error { return err } body := map[string]string{ - "content": content, + "text": content, "source_type": sourceType, } - data, err := c.Post("/ingest", body) + data, err := c.Post("/ingest/text", body) if err != nil { return err } diff --git a/packages/core/alayaos_core/schemas/tree.py b/packages/core/alayaos_core/schemas/tree.py index bba2bab..e8953de 100644 --- a/packages/core/alayaos_core/schemas/tree.py +++ b/packages/core/alayaos_core/schemas/tree.py @@ -3,10 +3,12 @@ import uuid from datetime import datetime -from pydantic import BaseModel, field_validator +from pydantic import BaseModel, ConfigDict, field_validator class TreeNodeResponse(BaseModel): + model_config = ConfigDict(from_attributes=True) + id: uuid.UUID path: str node_type: str diff --git a/packages/core/alayaos_core/services/ask.py b/packages/core/alayaos_core/services/ask.py index f81a11c..ca5824b 100644 --- a/packages/core/alayaos_core/services/ask.py +++ b/packages/core/alayaos_core/services/ask.py @@ -96,8 +96,11 @@ async def ask( max_tokens=settings.ASK_MAX_OUTPUT_TOKENS, ) + # Collect valid IDs from all evidence (including chunk-sourced with entity_id/claim_id) valid_entity_ids = {u.source_id for u in evidence if u.source_type == "entity"} + valid_entity_ids |= {u.entity_id for u in evidence if u.entity_id is not None} valid_claim_ids = {u.source_id for u in evidence if u.source_type == "claim"} + valid_claim_ids |= {u.claim_id for u in evidence if u.claim_id is not None} validated_citations = [] for c in response.citations: if c.entity_id and c.entity_id not in valid_entity_ids: