main.go

package main

import (
	"context"
	"log"
	"os"

	"sql-agent/api/server"
	"sql-agent/business/agent"
	"sql-agent/database"
)

func main() {
	// Initialize database
	db, err := database.Connect(
		os.Getenv("DATABASE_DSN_STRING"))
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// Initialize LLM models
	executor, err := agent.New(
		context.Background(),
		os.Getenv("GEMINI_API_KEY"),
		"gemini-2.0-flash",
		"Alwaleed98/walphi-sql",
		db)
	if err != nil {
		log.Fatal(err)
	}

	// Initialize and start server
	srv := server.New(executor)
	log.Println("Starting HTTP Server")
	if err := srv.Start(":8080"); err != nil {
		log.Fatal(err)
	}
}