mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-10 06:45:28 -05:00
We'll soon be needing a more feature-complete external API. To make way for this, I'm moving some files around so: - We can more easily create new versions of our external API - The file structure of our internal API is more homogeneous These changes are quite opinionated, but IMO in any case they're better than the chaotic structure we have now. ### Changes 🏗️ - Move `backend/server` -> `backend/api` - Move `backend/server/routers` + `backend/server/v2` -> `backend/api/features` - Change absolute sibling imports to relative imports - Move `backend/server/v2/AutoMod` -> `backend/executor/automod` - Combine `backend/server/routers/analytics_*test.py` -> `backend/api/features/analytics_test.py` - Sort OpenAPI spec file ### 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: - CI tests - [x] Clicking around in the app -> no obvious breakage
78 lines
1.7 KiB
Python
78 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.data.block import BlockInfo
|
|
from backend.integrations.providers import ProviderName
|
|
from backend.util.models import Pagination
|
|
|
|
FilterType = Literal[
|
|
"blocks",
|
|
"integrations",
|
|
"marketplace_agents",
|
|
"my_agents",
|
|
]
|
|
|
|
BlockType = 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):
|
|
otto_suggestions: list[str]
|
|
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
|