f9f6bc01fd
CI / test (push) Canceled after 0s
## Summary epic r04 - begin refactor ## Change Type - [x] Cowork feature - [ ] Bug fix - [ ] Core AI contribution - [ ] Test / hardening - [ ] Performance - [ ] Documentation ## Related Work Cowork Task: Core Repo: http://34.143.229.138/gitea-admin/fsg-ai-core-assets Core AI Issue: Core Task: Related PR: ## Scope What is intentionally included? What is intentionally NOT included? ## Validation - [ ] Unit tests - [ ] Integration tests - [ ] Manual verification - [ ] Regression check Commands / evidence: ## Security Impact Permission / credential / network / customer data impact: ## Compatibility - [ ] No breaking change - [ ] Breaking change documented ## Reviewer Notes Anything Cowork reviewers should pay attention to. --------- Co-authored-by: Anh Tran Nguyen Minh <anhtnm1@fpt.com> Co-authored-by: Huong Le Thi Thien <huongltt35@fpt.com> Co-authored-by: Nam Pham Dinh Thanh <nampdt@fpt.com> Co-authored-by: Vu Dam Tuan <vudt15@fpt.com> Co-authored-by: Hiep Ha Van <hiephv3@fpt.com> Co-authored-by: Lam Hoang Van <lamhv7@fpt.com> Reviewed-on: #7 Co-authored-by: Duy Le Huu <duylh19@fpt.com>
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""Member B work package: search_project_knowledge."""
|
|
|
|
# ruff: noqa: UP045 -- Optional keeps Pydantic model evaluation compatible with Python 3.9.
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Literal, Optional
|
|
|
|
from pydantic import Field
|
|
|
|
from ..foundation import ContractModel, SourceCitation, ToolTemplate
|
|
|
|
|
|
class KnowledgeSearchInput(ContractModel):
|
|
"""Tham số vào của ``search_project_knowledge``: project và câu truy vấn.
|
|
|
|
``top_k`` chặn trong 1..20 để một lượt tra không nuốt hết cửa sổ ngữ cảnh.
|
|
"""
|
|
project_id: str = Field(min_length=1, max_length=128)
|
|
query: str = Field(min_length=2, max_length=1000)
|
|
detail: Literal["summary", "standard", "full"] = "standard"
|
|
top_k: int = Field(default=5, ge=1, le=20)
|
|
language: Optional[Literal["en", "ja", "vi"]] = None
|
|
cursor: Optional[str] = Field(default=None, max_length=2048)
|
|
|
|
|
|
class KnowledgeItem(ContractModel):
|
|
"""Một đoạn tài liệu khớp truy vấn: trích đoạn, điểm số 0..1 và nguồn."""
|
|
document_id: str
|
|
chunk_id: str
|
|
title: str
|
|
excerpt: str
|
|
score: float = Field(ge=0, le=1)
|
|
source: SourceCitation
|
|
|
|
|
|
class KnowledgeSearchOutput(ContractModel):
|
|
"""Kết quả trả về: danh sách đoạn khớp đã xếp hạng, mỗi đoạn kèm trích dẫn nguồn."""
|
|
correlation_id: str
|
|
project_id: str
|
|
query: str
|
|
items: tuple[KnowledgeItem, ...]
|
|
truncated: bool
|
|
returned: int = Field(ge=0)
|
|
remaining: int = Field(ge=0)
|
|
next_cursor: Optional[str] = None
|
|
|
|
|
|
def _handle(arguments: ContractModel, provider: Any) -> dict[str, Any]:
|
|
"""Kiểm tham số theo schema rồi chuyển sang provider đã được cấp cho lượt gọi này."""
|
|
request = KnowledgeSearchInput.model_validate(arguments)
|
|
return provider.search_knowledge(**request.model_dump())
|
|
|
|
|
|
TOOL = ToolTemplate(
|
|
name="search_project_knowledge",
|
|
description=(
|
|
"Searches approved knowledge for one authorized project and returns ranked excerpts with "
|
|
"pinned citations. Use for requirements, design notes, or runbooks when no exact issue is "
|
|
"known. Do not use for issue details or Git change review."
|
|
),
|
|
input_model=KnowledgeSearchInput,
|
|
output_model=KnowledgeSearchOutput,
|
|
handler=_handle,
|
|
)
|