mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-18 22:42:54 +00:00
feed806489
* support groupchat, other QOL fixes * remove gallery success toast * Fix #1328. Add CSVLoader component and related support for rendering CSV files. Add download link in the modal for appropriate file types including CSV, Code, and PDF. * add name and description field to session datamodel * Update website/blog/2023-12-01-AutoGenStudio/index.mdx Co-authored-by: Chi Wang <wang.chi@microsoft.com> * sanitize llmconfig, remove additional fields * improve models UX, only modify models from model tab. * readme updates * improve db defaults * improve ui hover behavior and add note on models * general qol updats * add support for returning summary_method * use ant design tables * icon and layout updates * css and layout updates * readme updates and QOL updates * fix bug where empty string is used as apikey #1415 * add speaker selection to UI #1373 * Fixed a bug that localAgent updates were not synchronized between GroupChatFlowSpecView and AgentFlowSpecView. * Fixed a bug in Agent Specification Modal that caused localAgent updates to remain in state when closing a modal other than onOk. * Fixed a bug that the updated Agent Specification Modal was not saved when the content of FlowConfigViewer Modal was changed after the Agent Specification Modal was updated when an updatedFlowConfig was created using localFlowConfig. * add version to package * remove sample key * early support for versions table and testing models * Add support for testing model when created #1404 * remove unused imports, qol updates * fix bug on workflowmanager * make file_name optional in skills datamodel * update instructions on models * fix errors from merge conflict with main * santize workflow before download * add support for editing skills in full fledged editor (monaco) #1442 * fix merge artifacts * Fix build command for windows Replaced && to & to continue execution when the 'ui' folder doesn't exist and also suppressed error "The system cannot find the file specified." * Fix setup instructions The config file starts with a dot (according to gatsby-config.ts). * Throw error if env file doesn't exist Otherwise the app will not work (issue very hard to trace) * version bump * formattin gupdates * formatting updates * Show warning instead of error if env config file doesn't exist Fix: https://github.com/microsoft/autogen/pull/1475#issuecomment-1918114520 * add rel noopener to a tags * formating updates * remove double section in readme. * update dev readme * format update * add default autoreply to agent config datamodel * add check for empty messages list * improve groupchat behavior, add sender to list of agents * update system_message defaults to fit autogen default system message #1474 * simplify response from test_model to only return content, fix serialization issue in #1404 * readme and other formatting updates * add support for showing temp and default auto reply #1521 * formatting updates * formating and other updates --------- Co-authored-by: Paul Retherford <paul@scanpower.com> Co-authored-by: Chi Wang <wang.chi@microsoft.com> Co-authored-by: junkei_okinawa <ceazy.x2.okinawan@gmail.com> Co-authored-by: Christopher Pereira <kripper@imatronix.com>
303 lines
8.3 KiB
Python
303 lines
8.3 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from typing import Any, Callable, Dict, List, Literal, Optional, Union
|
|
from pydantic.dataclasses import dataclass
|
|
from dataclasses import asdict, field
|
|
|
|
|
|
@dataclass
|
|
class Message(object):
|
|
user_id: str
|
|
role: str
|
|
content: str
|
|
root_msg_id: Optional[str] = None
|
|
msg_id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
personalize: Optional[bool] = False
|
|
ra: Optional[str] = None
|
|
code: Optional[str] = None
|
|
metadata: Optional[Any] = None
|
|
session_id: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
if self.msg_id is None:
|
|
self.msg_id = str(uuid.uuid4())
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class Skill(object):
|
|
title: str
|
|
content: str
|
|
file_name: Optional[str] = None
|
|
id: Optional[str] = None
|
|
description: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
user_id: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.user_id is None:
|
|
self.user_id = "default"
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
return result
|
|
|
|
|
|
# web api data models
|
|
|
|
|
|
# autogenflow data models
|
|
@dataclass
|
|
class Model:
|
|
"""Data model for Model Config item in LLMConfig for AutoGen"""
|
|
|
|
model: str
|
|
api_key: Optional[str] = None
|
|
base_url: Optional[str] = None
|
|
api_type: Optional[str] = None
|
|
api_version: Optional[str] = None
|
|
id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
user_id: Optional[str] = None
|
|
description: Optional[str] = None
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
return result
|
|
|
|
def __post_init__(self):
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.user_id is None:
|
|
self.user_id = "default"
|
|
|
|
|
|
@dataclass
|
|
class LLMConfig:
|
|
"""Data model for LLM Config for AutoGen"""
|
|
|
|
config_list: List[Any] = field(default_factory=list)
|
|
temperature: float = 0
|
|
cache_seed: Optional[Union[int, None]] = None
|
|
timeout: Optional[int] = None
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
result["config_list"] = [c.dict() for c in self.config_list]
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class AgentConfig:
|
|
"""Data model for Agent Config for AutoGen"""
|
|
|
|
name: str
|
|
llm_config: Optional[Union[LLMConfig, bool]] = False
|
|
human_input_mode: str = "NEVER"
|
|
max_consecutive_auto_reply: int = 10
|
|
system_message: Optional[str] = None
|
|
is_termination_msg: Optional[Union[bool, str, Callable]] = None
|
|
code_execution_config: Optional[Union[bool, str, Dict[str, Any]]] = None
|
|
default_auto_reply: Optional[str] = ""
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
if isinstance(result["llm_config"], LLMConfig):
|
|
result["llm_config"] = result["llm_config"].dict()
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class AgentFlowSpec:
|
|
"""Data model to help flow load agents from config"""
|
|
|
|
type: Literal["assistant", "userproxy"]
|
|
config: AgentConfig
|
|
id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
user_id: Optional[str] = None
|
|
skills: Optional[Union[None, List[Skill]]] = None
|
|
description: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
if self.user_id is None:
|
|
self.user_id = "default"
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class GroupChatConfig:
|
|
"""Data model for GroupChat Config for AutoGen"""
|
|
|
|
agents: List[AgentFlowSpec] = field(default_factory=list)
|
|
admin_name: str = "Admin"
|
|
messages: List[Dict] = field(default_factory=list)
|
|
max_round: Optional[int] = 10
|
|
admin_name: Optional[str] = "Admin"
|
|
speaker_selection_method: Optional[str] = "auto"
|
|
allow_repeat_speaker: Optional[Union[bool, List[AgentConfig]]] = True
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
result["agents"] = [a.dict() for a in self.agents]
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class GroupChatFlowSpec:
|
|
"""Data model to help flow load agents from config"""
|
|
|
|
type: Literal["groupchat"]
|
|
config: AgentConfig = field(default_factory=AgentConfig)
|
|
groupchat_config: Optional[GroupChatConfig] = field(default_factory=GroupChatConfig)
|
|
id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
user_id: Optional[str] = None
|
|
description: Optional[str] = None
|
|
skills: Optional[Union[None, List[Skill]]] = None
|
|
|
|
def __post_init__(self):
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
if self.user_id is None:
|
|
self.user_id = "default"
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
# result["config"] = self.config.dict()
|
|
# result["groupchat_config"] = self.groupchat_config.dict()
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class AgentWorkFlowConfig:
|
|
"""Data model for Flow Config for AutoGen"""
|
|
|
|
name: str
|
|
description: str
|
|
sender: AgentFlowSpec
|
|
receiver: Union[AgentFlowSpec, GroupChatFlowSpec]
|
|
type: Literal["twoagents", "groupchat"] = "twoagents"
|
|
id: Optional[str] = None
|
|
user_id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
# how the agent message summary is generated. last: only last message is used, none: no summary, llm: use llm to generate summary
|
|
summary_method: Optional[Literal["last", "none", "llm"]] = "last"
|
|
|
|
def init_spec(self, spec: Dict):
|
|
"""initialize the agent spec"""
|
|
if not isinstance(spec, dict):
|
|
spec = spec.dict()
|
|
if spec["type"] == "groupchat":
|
|
return GroupChatFlowSpec(**spec)
|
|
else:
|
|
return AgentFlowSpec(**spec)
|
|
|
|
def __post_init__(self):
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
self.sender = self.init_spec(self.sender)
|
|
self.receiver = self.init_spec(self.receiver)
|
|
if self.user_id is None:
|
|
self.user_id = "default"
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
result["sender"] = self.sender.dict()
|
|
result["receiver"] = self.receiver.dict()
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class Session(object):
|
|
"""Data model for AutoGen Chat Session"""
|
|
|
|
user_id: str
|
|
id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
flow_config: AgentWorkFlowConfig = None
|
|
name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
result["flow_config"] = self.flow_config.dict()
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class Gallery(object):
|
|
"""Data model for Gallery Item"""
|
|
|
|
session: Session
|
|
messages: List[Message]
|
|
tags: List[str]
|
|
id: Optional[str] = None
|
|
timestamp: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
if self.timestamp is None:
|
|
self.timestamp = datetime.now().isoformat()
|
|
if self.id is None:
|
|
self.id = str(uuid.uuid4())
|
|
|
|
def dict(self):
|
|
result = asdict(self)
|
|
return result
|
|
|
|
|
|
@dataclass
|
|
class ChatWebRequestModel(object):
|
|
"""Data model for Chat Web Request for Web End"""
|
|
|
|
message: Message
|
|
flow_config: AgentWorkFlowConfig
|
|
|
|
|
|
@dataclass
|
|
class DeleteMessageWebRequestModel(object):
|
|
user_id: str
|
|
msg_id: str
|
|
session_id: Optional[str] = None
|
|
|
|
|
|
@dataclass
|
|
class DBWebRequestModel(object):
|
|
user_id: str
|
|
msg_id: Optional[str] = None
|
|
session: Optional[Session] = None
|
|
skill: Optional[Skill] = None
|
|
tags: Optional[List[str]] = None
|
|
agent: Optional[AgentFlowSpec] = None
|
|
workflow: Optional[AgentWorkFlowConfig] = None
|
|
model: Optional[Model] = None
|