mirror of
https://github.com/csunny/DB-GPT.git
synced 2026-01-25 23:04:03 +00:00
ed702db9bc
1.AgentHub Module complete 2.New ChatAgent Mode complete 3.MetaData develop，auto update ddl
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
from typing import TypedDict, Optional, Dict, List
|
|
from dataclasses import dataclass
|
|
from pydantic import BaseModel, Field
|
|
from typing import TypeVar, Generic, Any
|
|
|
|
T = TypeVar('T')
|
|
|
|
class PagenationFilter(BaseModel, Generic[T]):
|
|
page_index: int = 1
|
|
page_size: int = 20
|
|
filter: T = None
|
|
|
|
class PagenationResult(BaseModel, Generic[T]):
|
|
page_index: int = 1
|
|
page_size: int = 20
|
|
total_page: int = 0
|
|
total_row_count: int = 0
|
|
datas: List[T] = []
|
|
|
|
def to_dic(self):
|
|
data_dicts =[]
|
|
for item in self.datas:
|
|
data_dicts.append(item.__dict__)
|
|
return {
|
|
'page_index': self.page_index,
|
|
'page_size': self.page_size,
|
|
'total_page': self.total_page,
|
|
'total_row_count': self.total_row_count,
|
|
'datas': data_dicts
|
|
}
|
|
|
|
@dataclass
|
|
class PluginHubFilter(BaseModel):
|
|
name: str
|
|
description: str
|
|
author: str
|
|
email: str
|
|
type: str
|
|
version: str
|
|
storage_channel: str
|
|
storage_url: str
|
|
|
|
|
|
@dataclass
|
|
class MyPluginFilter(BaseModel):
|
|
tenant: str
|
|
user_code: str
|
|
user_name: str
|
|
name: str
|
|
file_name: str
|
|
type: str
|
|
version: str
|
|
|
|
|
|
class PluginHubParam(BaseModel):
|
|
channel: Optional[str] = Field("git", description="Plugin storage channel")
|
|
url: Optional[str] = Field("https://github.com/eosphoros-ai/DB-GPT-Plugins.git", description="Plugin storage url")
|
|
branch: Optional[str] = Field("main", description="github download branch", nullable=True)
|
|
authorization: Optional[str] = Field(None, description="github download authorization", nullable=True)
|
|
|
|
|