+4
-7
@@ -6,11 +6,10 @@ import (
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
searchagent "aichat/agents/search"
|
||||
sqlquery "aichat/agents/sql"
|
||||
"aichat/config"
|
||||
"aichat/conversation"
|
||||
"aichat/llm"
|
||||
"aichat/toolmanager"
|
||||
"aichat/toolrouter"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -19,19 +18,17 @@ import (
|
||||
type Server struct {
|
||||
cfg *config.Config
|
||||
aiState *llm.State
|
||||
searchState *searchagent.State
|
||||
sqlState *sqlquery.State
|
||||
toolManager *toolmanager.Manager
|
||||
toolRouterState *toolrouter.State
|
||||
store *conversation.Store
|
||||
router *gin.Engine
|
||||
}
|
||||
|
||||
func New(cfg *config.Config, aiState *llm.State, searchState *searchagent.State, sqlState *sqlquery.State, toolRouterState *toolrouter.State, store *conversation.Store) *Server {
|
||||
func New(cfg *config.Config, aiState *llm.State, toolManager *toolmanager.Manager, toolRouterState *toolrouter.State, store *conversation.Store) *Server {
|
||||
s := &Server{
|
||||
cfg: cfg,
|
||||
aiState: aiState,
|
||||
searchState: searchState,
|
||||
sqlState: sqlState,
|
||||
toolManager: toolManager,
|
||||
toolRouterState: toolRouterState,
|
||||
store: store,
|
||||
router: gin.Default(),
|
||||
|
||||
Reference in New Issue
Block a user