switch to gemini model, update supabase_model managed agent query and update the flow (#10)

.python-version

@@ -1 +1 @@
-3.11.6
+3.13

LLM/llm_models.py

@@ -3,10 +3,18 @@ from .models import LLMProviderType, LLMModelType
 from .llm_provider import LLMProvider
 
 
-orchestrator_model = LLMProvider(LLMProviderType.HF, LLMModelType.open_source.deepseek_v3_terminus).get_model()
+use_gemini = False
 
-supabase_model = LLMProvider(LLMProviderType.HF, LLMModelType.open_source.deepseek_v3_terminus).get_model()
-websearch_model = LLMProvider(LLMProviderType.HF, LLMModelType.open_source.qwen_80b_thinking).get_model()
+base_provider = LLMProviderType.GEMINI if use_gemini else LLMProviderType.HF
+
+base_model = LLMModelType.gemini.gemini_2_5_pro if use_gemini else LLMModelType.open_source.deepseek_v3_terminus
+
+
+
+orchestrator_model = LLMProvider(base_provider, base_model).get_model()
+
+supabase_model = LLMProvider(base_provider, base_model).get_model()
+websearch_model = LLMProvider(base_provider, base_model).get_model()
 
 context_compression_model = LLMProvider(LLMProviderType.OPENAI, LLMModelType.openai.gpt_5_1).get_model()
 metaprompting_model = LLMProvider(LLMProviderType.OPENAI, LLMModelType.openai.gpt_5_1).get_model()

LLM/models.py

@@ -17,6 +17,7 @@ class AnthropicModel(str, Enum):
 
 
 class GeminiModel(str, Enum):
+    gemini_3_pro_preview = "gemini-3-pro-preview"
     gemini_2_5_flash = "gemini-2.5-flash"
     gemini_2_5_pro = "gemini-2.5-pro"
 
@@ -29,7 +30,7 @@ class OSSModel(str, Enum):
     deepseek_r1 = "deepseek-ai/DeepSeek-R1"
     gpt_oss_120b = "openai/gpt-oss-120b"
     kimi_k2_thinking = "moonshotai/Kimi-K2-Thinking",
-    qwen_80b_thinking = 'Qwen/Qwen3-Next-80B-A3B-Thinking'
+    qwen_3_coder = 'Qwen/Qwen3-Coder-480B-A35B-Instruct'
 
 class EnumNamespace:
     """Proxy to expose Enum members as attribute-accessed .value strings."""

UI/render_employee_card.py

@@ -17,7 +17,8 @@ def _as_mapping(member: EmployeeTrainingPlan | Mapping[str, Any]) -> Mapping[str
     if hasattr(member, "model_dump"):
         return member.model_dump()
     return {
-        "employee_id": getattr(member, "employee_id", "Unknown employee"),
+        "employee_name": getattr(member, "employee_name", "Unknown employee"),
+        "role": getattr(member, "role", "Unknown role"),
         "skills_gaps": getattr(member, "skills_gaps", []),
         "training_plan": getattr(member, "training_plan", []),
     }
@@ -108,7 +109,7 @@ def _render_training_plan(resources: Iterable[Mapping[str, Any]] | None) -> str:
 
 def render_employee_card(member: EmployeeTrainingPlan | Mapping[str, Any]) -> str:
     data = _as_mapping(member)
-    name = html.escape(str(data.get("employee_id", "Unknown employee")))
+    name = html.escape(str(data.get("employee_name", "Unknown employee")))
     role = html.escape(str(data.get("role") or "Not provided"))
 
     skill_gap_html = _render_skill_gaps(data.get("skills_gaps"))

agents/orchestrator_agent/orchestrator_agent.py

@@ -19,7 +19,7 @@ metaprompting_agent = MetapromptingAgent()
 SUGGEST_METAPROMPT_IMPROVEMENTS = False
 
 @tool
-def final_answer_callback(project_id: str, project_name: str, budget:int, deadline: str, status: str, team: List[EmployeeTrainingPlan]) -> AnalysisResult:
+def final_answer_callback(project_id: int, project_name: str, budget:int, deadline: str, status: str, team: List[EmployeeTrainingPlan]) -> AnalysisResult:
     """
     Finalizes the training-plan analysis and hands structured data back to the app about the project, the team and the training plan.
     ⚠️ This MUST be the last call. Do not call it before all skill gaps and training items are collected.

agents/supabase_agent/get_supabase_agent_prompt.py

@@ -62,6 +62,8 @@ Each returns:
 
 Therefore: NEVER recompute gap logic client-side unless strictly necessary.
 
+When executing a function, execute it like this: execute_sql(query=.....), don't pass the query directly, use keyword args.
+
 ────────────────────────────────────
 WHEN TO SUGGEST NEW RPC
 ────────────────────────────────────

agents/websearch_agent/get_web_search_agent.py

@@ -1,5 +1,5 @@
 
-from smolagents import ToolCallingAgent, WebSearchTool
+from smolagents import ToolCallingAgent, DuckDuckGoSearchTool
 
 from LLM.llm_models import websearch_model
 
@@ -12,7 +12,7 @@ def get_web_search_agent(step_callbacks):
     if _web_search_agent is None:
         _web_search_agent = ToolCallingAgent(
             model=websearch_model, 
-            tools=[WebSearchTool()], 
+            tools=[DuckDuckGoSearchTool()], 
             step_callbacks=step_callbacks,
             name="web_search_agent", 
             description="An agent that searches for relevant online courses, certifications, and learning paths."

mcp_server/server.py

@@ -1,25 +0,0 @@
-from fastmcp import FastMCP
-from fastmcp.tools import Tool
-
-mcp = FastMCP('mcp_server')
-# Minimal MCP stdio server skeleton
-from .tools import (
-    get_team_members,
-    get_employees_skills,
-    get_project_requirements,
-    search_training_materials,
-    save_training_plan,
-    get_training_plans,
-)
-mcp.add_tool(Tool.from_function(get_team_members))
-mcp.add_tool(Tool.from_function(get_employees_skills))
-mcp.add_tool(Tool.from_function(get_project_requirements))
-mcp.add_tool(Tool.from_function(search_training_materials))
-mcp.add_tool(Tool.from_function(save_training_plan))
-mcp.add_tool(Tool.from_function(get_training_plans))
-def main():
-    # Placeholder for stdio server loop
-    mcp.run()
-
-if __name__ == "__main__":
-    main()

mcp_server/tools.py

@@ -1,21 +0,0 @@
-from typing import List, Dict, Any
-from models import TrainingPlan
-
-
-def get_team_members(project_id: str) -> List[str]:
-    raise NotImplementedError("Implement this later")
-
-def get_employees_skills(employee_ids: List[str]) -> Dict[str, Dict[str, int]]:
-    raise NotImplementedError("Implement this later")
-
-def get_project_requirements(project_id: str) -> Dict[str, int]:
-    raise NotImplementedError("Implement this later")
-
-def save_training_plan(training_plan: TrainingPlan) -> None:
-    raise NotImplementedError("Implement this later")
-
-def get_training_plans(project_id: str) -> List[TrainingPlan]:
-    raise NotImplementedError("Implement this later")
-
-
-

models.py

@@ -1,21 +1,20 @@
 from __future__ import annotations
 from pydantic import BaseModel
-from typing import List, Dict, Optional
-from datetime import datetime
+from typing import List,  Optional
 
 class Employee(BaseModel):
     id: str
     name: str
-    role: str
-    skills: Dict[str, int]  # skill name -> proficiency (0-100)
+    role: str   
+    department: str
+    level: str
+    project_id: List[str]
+    hourly_rate: Optional[float] = None
 
 class EmployeeSkillGap(BaseModel):
     employee_id: str
     skill: str
 
-class ProjectRequirement(BaseModel):
-    project_id: str
-    required_skills: Dict[str, int]  # skill -> desired proficiency
 
 class TrainingMaterial(BaseModel):
     id: str
@@ -25,16 +24,8 @@ class TrainingMaterial(BaseModel):
     cost: Optional[float] = None
     duration_hours: Optional[float] = None
 
-class TrainingPlan(BaseModel):
-    id: Optional[str] = None
-    project_id: str
-    employees: List[str]
-    recommended_materials: Dict[str, List[TrainingMaterial]]
-    total_cost: Optional[float] = None
-    created_at: Optional[datetime] = None
-
 class EmployeeTrainingPlan(BaseModel):
-    employee_id: str
+    employee_name: str
     role: str
     skills_gaps: List[EmployeeSkillGap]
     training_plan: List[TrainingMaterial]

pyproject.toml

@@ -6,7 +6,7 @@ version = "0.1.0"
 description = "SmolAgents-based HR skill-gap analysis and training planner app (EastSync)"
 authors = ["EastSync Team"]
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.13"
 dependencies = [
 	"pydantic>=2.0.0",
 	"gradio==6.0.0.dev1",
@@ -15,6 +15,7 @@ dependencies = [
 	"fastmcp>=2.13.1",
 	"ddgs>=9.9.1",
 	"repr>=0.3.1",
+	"typer>=0.20.0",
 ]
 
 [tool.uv]

response_sample.json

@@ -1,163 +1,143 @@
 {
-   "project_id":"2",
-   "project_name":"Cloud Migration Initiative",
-   "budget":180000,
-   "deadline":"2025-07-01",
-   "status":"Ongoing",
-   "team":[
-      {
-         "employee_id":"4",
-         "role":"DevOps Engineer",
-         "skills_gaps":[
-            
-         ],
-         "training_plan":[
-            
-         ]
-      },
-      {
-         "employee_id":"5",
-         "role":"Cloud Engineer",
-         "skills_gaps":[
-            {
-               "employee_id":"5",
-               "skill":"Terraform"
-            }
-         ],
-         "training_plan":[
-            {
-               "id":"terraform_1",
-               "title":"HashiCorp Terraform Associate Certification",
-               "url":"https://developer.hashicorp.com/certifications",
-               "skills_covered":[
-                  "Terraform"
-               ],
-               "cost":100.0,
-               "duration_hours":20.0
-            }
-         ]
-      },
-      {
-         "employee_id":"6",
-         "role":"Java Engineer",
-         "skills_gaps":[
-            {
-               "employee_id":"6",
-               "skill":"AWS"
-            },
-            {
-               "employee_id":"6",
-               "skill":"Docker"
-            },
-            {
-               "employee_id":"6",
-               "skill":"Terraform"
-            }
-         ],
-         "training_plan":[
-            {
-               "id":"aws_1",
-               "title":"AWS Skill Builder (Free Courses)",
-               "url":"https://skillbuilder.aws",
-               "skills_covered":[
-                  "AWS"
-               ],
-               "cost":0.0,
-               "duration_hours":25.0
-            },
-            {
-               "id":"docker_2",
-               "title":"Coursera Docker Specialization",
-               "url":"https://www.coursera.org/courses?query=docker",
-               "skills_covered":[
-                  "Docker"
-               ],
-               "cost":49.0,
-               "duration_hours":40.0
-            },
-            {
-               "id":"terraform_2",
-               "title":"Coursera Terraform Courses",
-               "url":"https://www.coursera.org/courses?query=terraform",
-               "skills_covered":[
-                  "Terraform"
-               ],
-               "cost":49.0,
-               "duration_hours":30.0
-            }
-         ]
-      },
-      {
-         "employee_id":"19",
-         "role":"Software Engineer",
-         "skills_gaps":[
-            {
-               "employee_id":"19",
-               "skill":"AWS"
-            },
-            {
-               "employee_id":"19",
-               "skill":"Docker"
-            },
-            {
-               "employee_id":"19",
-               "skill":"Terraform"
-            }
-         ],
-         "training_plan":[
-            {
-               "id":"aws_3",
-               "title":"AWS Solutions Architect - Associate",
-               "url":"https://aws.amazon.com/training/",
-               "skills_covered":[
-                  "AWS"
-               ],
-               "cost":150.0,
-               "duration_hours":40.0
-            },
-            {
-               "id":"docker_1",
-               "title":"Docker Certified Associate (DCA) Preparation by ONLC",
-               "url":"https://www.onlc.com/docker-training-classes.htm",
-               "skills_covered":[
-                  "Docker"
-               ],
-               "cost":2000.0,
-               "duration_hours":32.0
-            },
-            {
-               "id":"terraform_3",
-               "title":"Edstellar Terraform CorporateTraining",
-               "url":"https://www.edstellar.com/course/terraform-training",
-               "skills_covered":[
-                  "Terraform"
-               ],
-               "cost":2000.0,
-               "duration_hours":20.0
-            }
-         ]
-      },
-      {
-         "employee_id":"20",
-         "role":"Cloud Architect",
-         "skills_gaps":[
-            {
-               "employee_id":"20",
-               "skill":"Docker"
-            }
-         ],
-         "training_plan":[
-            {
-               "id":"docker_1",
-               "title":"Docker Certified Associate (DCA) Preparation by ONLC",
-               "url":"https://www.onlc.com/docker-training-classes.htm",
-               "skills_covered":[
-                  "Docker"
-               ],
-               "cost":2000.0,
-               "duration_hours":32.0
-            }
-         ]
-      }
-   ]
+  "project_id": 2,
+  "project_name": "Cloud Migration Initiative",
+  "budget": 180000,
+  "deadline": "2025-07-01",
+  "status": "Ongoing",
+  "team": [
+    {
+      "employee_name": "Emily Davis",
+      "role": "DevOps Engineer",
+      "skills_gaps": [],
+      "training_plan": []
+    },
+    {
+      "employee_name": "Michael Brown",
+      "role": "Cloud Engineer",
+      "skills_gaps": [
+        { "skill": "Terraform" }
+      ],
+      "training_plan": [
+        {
+          "id": "T01",
+          "title": "Terraform Online Training and Certification Course",
+          "url": "https://secappslearning.com/course/terraform-online-training-and-certification-course",
+          "skills_covered": ["Terraform"],
+          "cost": 24999,
+          "duration_hours": 35
+        }
+      ]
+    },
+    {
+      "employee_name": "Daniel Wilson",
+      "role": "Java Engineer",
+      "skills_gaps": [
+        { "skill": "AWS" },
+        { "skill": "Docker" },
+        { "skill": "Terraform" }
+      ],
+      "training_plan": [
+        {
+          "id": "A01",
+          "title": "AWS Training and Certification",
+          "url": "https://mindmajix.com/aws-training",
+          "skills_covered": ["AWS"],
+          "cost": 0,
+          "duration_hours": 25
+        },
+        {
+          "id": "D01",
+          "title": "Docker for the Absolute Beginner",
+          "url": "https://www.coursera.org/learn/docker-for-the-absolute-beginner",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 16
+        },
+        {
+          "id": "D02",
+          "title": "Docker - Introducing Docker Essentials, Containers, and more",
+          "url": "https://www.udemy.com/course/docker-introducing-docker-essentials-containers-and-more/",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 2.5
+        },
+        {
+          "id": "T01",
+          "title": "Terraform Online Training and Certification Course",
+          "url": "https://secappslearning.com/course/terraform-online-training-and-certification-course",
+          "skills_covered": ["Terraform"],
+          "cost": 24999,
+          "duration_hours": 35
+        }
+      ]
+    },
+    {
+      "employee_name": "Jacob Collins",
+      "role": "Software Engineer",
+      "skills_gaps": [
+        { "skill": "AWS" },
+        { "skill": "Docker" },
+        { "skill": "Terraform" }
+      ],
+      "training_plan": [
+        {
+          "id": "A01",
+          "title": "AWS Training and Certification",
+          "url": "https://mindmajix.com/aws-training",
+          "skills_covered": ["AWS"],
+          "cost": 0,
+          "duration_hours": 25
+        },
+        {
+          "id": "D01",
+          "title": "Docker for the Absolute Beginner",
+          "url": "https://www.coursera.org/learn/docker-for-the-absolute-beginner",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 16
+        },
+        {
+          "id": "D02",
+          "title": "Docker - Introducing Docker Essentials, Containers, and more",
+          "url": "https://www.udemy.com/course/docker-introducing-docker-essentials-containers-and-more/",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 2.5
+        },
+        {
+          "id": "T01",
+          "title": "Terraform Online Training and Certification Course",
+          "url": "https://secappslearning.com/course/terraform-online-training-and-certification-course",
+          "skills_covered": ["Terraform"],
+          "cost": 24999,
+          "duration_hours": 35
+        }
+      ]
+    },
+    {
+      "employee_name": "Chloe Morgan",
+      "role": "Cloud Architect",
+      "skills_gaps": [
+        { "skill": "Docker" }
+      ],
+      "training_plan": [
+        {
+          "id": "D01",
+          "title": "Docker for the Absolute Beginner",
+          "url": "https://www.coursera.org/learn/docker-for-the-absolute-beginner",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 16
+        },
+        {
+          "id": "D02",
+          "title": "Docker - Introducing Docker Essentials, Containers, and more",
+          "url": "https://www.udemy.com/course/docker-introducing-docker-essentials-containers-and-more/",
+          "skills_covered": ["Docker"],
+          "cost": 0,
+          "duration_hours": 2.5
+        }
+      ]
+    }
+  ]
 }