mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-04-08 03:00:28 -04:00
c7cdb40c5b
### Changes 🏗️ - Add materialized view for suggested blocks - Make `search` in builder accept comma separated filter list in query - Remove Otto suggestions - Use hybrid search for blocks search in builder - Exclude `AgentExecutorBlocks` from builder - Remove `Block` suffix from builder items ### Checklist 📋 #### For code changes: - [x] I have clearly listed my changes in the PR description - [x] I have made a test plan - [x] I have tested my changes according to the test plan: - [x] Materialized view function works (when running manually) - [x] Higher execution count blocks are shown first in "suggested blocks" (uses materialized view) - [x] Hybrid search works - [x] `AgentExecutorBlocks` doesn't appear on search results and in blocks list - [x] `Block` suffix isn't displayed on blocks names in builder items --------- Co-authored-by: Reinier van der Leer <pwuts@agpt.co>
77 lines
1.7 KiB
Python
77 lines
1.7 KiB
Python
from typing import Literal
|
|
|
|
from pydantic import BaseModel
|
|
|
|
import backend.api.features.library.model as library_model
|
|
import backend.api.features.store.model as store_model
|
|
from backend.blocks._base import BlockInfo
|
|
from backend.integrations.providers import ProviderName
|
|
from backend.util.models import Pagination
|
|
|
|
FilterType = Literal[
|
|
"blocks",
|
|
"integrations",
|
|
"marketplace_agents",
|
|
"my_agents",
|
|
]
|
|
|
|
BlockTypeFilter = Literal["all", "input", "action", "output"]
|
|
|
|
|
|
class SearchEntry(BaseModel):
|
|
search_query: str | None = None
|
|
filter: list[FilterType] | None = None
|
|
by_creator: list[str] | None = None
|
|
search_id: str | None = None
|
|
|
|
|
|
# Suggestions
|
|
class SuggestionsResponse(BaseModel):
|
|
recent_searches: list[SearchEntry]
|
|
providers: list[ProviderName]
|
|
top_blocks: list[BlockInfo]
|
|
|
|
|
|
# All blocks
|
|
class BlockCategoryResponse(BaseModel):
|
|
name: str
|
|
total_blocks: int
|
|
blocks: list[BlockInfo]
|
|
|
|
model_config = {"use_enum_values": False} # Use enum names like "AI"
|
|
|
|
|
|
# Input/Action/Output and see all for block categories
|
|
class BlockResponse(BaseModel):
|
|
blocks: list[BlockInfo]
|
|
pagination: Pagination
|
|
|
|
|
|
# Providers
|
|
class Provider(BaseModel):
|
|
name: ProviderName
|
|
description: str
|
|
integration_count: int
|
|
|
|
|
|
class ProviderResponse(BaseModel):
|
|
providers: list[Provider]
|
|
pagination: Pagination
|
|
|
|
|
|
class SearchResponse(BaseModel):
|
|
items: list[BlockInfo | library_model.LibraryAgent | store_model.StoreAgent]
|
|
search_id: str
|
|
total_items: dict[FilterType, int]
|
|
pagination: Pagination
|
|
|
|
|
|
class CountResponse(BaseModel):
|
|
all_blocks: int
|
|
input_blocks: int
|
|
action_blocks: int
|
|
output_blocks: int
|
|
integrations: int
|
|
marketplace_agents: int
|
|
my_agents: int
|