wehub-resource-sync
bcbd1bdb22
Integ / changes (push) Has been skipped
Pre-commit / pre-commit (push) Failing after 1s
CLI exit codes / changes (push) Has been skipped
Test (Install) / changes (push) Has been skipped
Test (Python) / changes (push) Has been skipped
Test (TypeScript) / changes (push) Has been skipped
CLI exit codes / cli-gate (push) Has been cancelled
Test (Install) / test-install-gate (push) Has been cancelled
Integ / integ-gate (push) Has been cancelled
Test (Python) / test-python-gate (push) Has been cancelled
Test (TypeScript) / test-typescript-gate (push) Has been cancelled
Test (Install) / python-minimal (3.12) (push) Has been cancelled
Test (Install) / python-minimal (3.11) (push) Has been cancelled
Test (Install) / python-extra (agno, mirage.agents.agno) (push) Has been cancelled
Test (Install) / python-extra (chroma, mirage.resource.chroma) (push) Has been cancelled
Test (Install) / python-extra (pdf, mirage.core.filetype.pdf) (push) Has been cancelled
Integ / integ (push) Has been cancelled
Integ / integ-database (push) Has been cancelled
Integ / integ-database-ts (push) Has been cancelled
Integ / integ-data (push) Has been cancelled
Integ / integ-ssh (push) Has been cancelled
Integ / integ-ssh-ts (push) Has been cancelled
Test (Python) / audit (push) Has been cancelled
Test (TypeScript) / test (push) Has been cancelled
Test (TypeScript) / python-fs-shim (push) Has been cancelled
CLI exit codes / Python CLI (push) Has been cancelled
CLI exit codes / TypeScript CLI (push) Has been cancelled
CLI exit codes / Cross-language snapshot interop (push) Has been cancelled
Test (Python) / test (push) Has been cancelled
Test (Python) / import-isolation (deepagents, openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Python) / import-isolation (deepagents, pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Integ / integ-ts (push) Has been cancelled
Integ / integ-fuse (push) Has been cancelled
Test (Install) / python-extra (databricks, mirage.resource.databricks_volume) (push) Has been cancelled
Test (Install) / python-extra (deepagents, mirage.agents.langchain) (push) Has been cancelled
Test (Install) / python-extra (email, mirage.resource.email) (push) Has been cancelled
Test (Install) / python-extra (fuse, mirage.fuse.mount) (push) Has been cancelled
Test (Install) / python-extra (hdf5, mirage.core.filetype.hdf5) (push) Has been cancelled
Test (Install) / python-extra (hf, mirage.resource.hf_buckets) (push) Has been cancelled
Test (Install) / python-extra (lancedb, mirage.resource.lancedb) (push) Has been cancelled
Test (Install) / python-extra (langfuse, mirage.resource.langfuse) (push) Has been cancelled
Test (Install) / python-extra (mongodb, mirage.resource.mongodb) (push) Has been cancelled
Test (Install) / python-extra (nextcloud, mirage.resource.nextcloud) (push) Has been cancelled
Test (Install) / python-extra (openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Install) / python-extra (openhands, mirage.agents.openhands, 3.12) (push) Has been cancelled
Test (Install) / python-extra (parquet, mirage.core.filetype.parquet) (push) Has been cancelled
Test (Install) / python-extra (postgres, mirage.resource.postgres) (push) Has been cancelled
Test (Install) / python-extra (pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Test (Install) / python-extra (qdrant, mirage.resource.qdrant) (push) Has been cancelled
Test (Install) / python-extra (redis, mirage.resource.redis) (push) Has been cancelled
Test (Install) / python-extra (s3, mirage.resource.s3) (push) Has been cancelled
Test (Install) / python-extra (ssh, mirage.resource.ssh) (push) Has been cancelled
Test (Install) / ts-minimal (push) Has been cancelled
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from mirage.cache.index import IndexCacheStore
|
|
from mirage.commands.builtin.utils.paths import default_paths
|
|
from mirage.commands.registry import command
|
|
from mirage.commands.spec import SPECS
|
|
from mirage.core.dify import search as search_core
|
|
from mirage.core.dify.glob import resolve_glob
|
|
from mirage.io.types import ByteSource, IOResult
|
|
from mirage.types import PathSpec
|
|
from mirage.utils.key_prefix import mount_prefix_of
|
|
|
|
|
|
def is_mount_root(path: PathSpec) -> bool:
|
|
root = mount_prefix_of(path.virtual, path.resource_path).rstrip("/") or "/"
|
|
value = path.virtual.rstrip("/") or "/"
|
|
return value == "/" or value == root
|
|
|
|
|
|
@command("search", resource="dify", spec=SPECS["search"])
|
|
async def search(
|
|
accessor,
|
|
paths: list[PathSpec],
|
|
*texts: str,
|
|
method: str = "semantic",
|
|
top_k: str | int = 10,
|
|
threshold: str | float = 0.0,
|
|
index: IndexCacheStore | None = None,
|
|
cwd: PathSpec | None = None,
|
|
**_extra: object,
|
|
) -> tuple[ByteSource | None, IOResult]:
|
|
if not texts:
|
|
raise ValueError("search: query is required")
|
|
query = texts[0]
|
|
target_paths = default_paths(paths, cwd)
|
|
mount_prefix = mount_prefix_of(
|
|
target_paths[0].virtual,
|
|
target_paths[0].resource_path) if target_paths else ""
|
|
if any(is_mount_root(path) for path in target_paths):
|
|
resolved_paths: list[PathSpec] = []
|
|
else:
|
|
resolved_paths = await resolve_glob(accessor, target_paths, index)
|
|
output = await search_core.search_segments(accessor,
|
|
query,
|
|
resolved_paths,
|
|
index,
|
|
method=method,
|
|
top_k=int(top_k),
|
|
threshold=float(threshold),
|
|
mount_prefix=mount_prefix)
|
|
return output, IOResult()
|