From fbec5f466cfdb2f74b937be80900ee62e274e780 Mon Sep 17 00:00:00 2001
From: sanosh <sanosh@vizsatizs-MacBook-Pro.local>
Date: Mon, 18 Nov 2024 13:38:21 +0530
Subject: [PATCH 1/3] solving the build issue #61

---
 flo_ai/router/flo_linear.py     | 5 +++--
 flo_ai/router/flo_llm_router.py | 3 +++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/flo_ai/router/flo_linear.py b/flo_ai/router/flo_linear.py
index 6d5f8b82..cdc8b20e 100644
--- a/flo_ai/router/flo_linear.py
+++ b/flo_ai/router/flo_linear.py
@@ -61,10 +61,11 @@ def create(session: FloSession, name: str, team: FloTeam):
         return FloLinear.Builder(session=session, name=name, flo_team=team).build()
 
     class Builder:
-        def __init__(self, session: FloSession, name: str, flo_team: FloTeam) -> None:
+        def __init__(self, session: FloSession, name: str, flo_team: FloTeam, model_nick_name: str = 'default') -> None:
             self.name = name
             self.session = session
             self.team = flo_team
+            self.model_name = model_nick_name
 
         def build(self):
-            return FloLinear(self.session, self.name, self.team)
+            return FloLinear(self.session, self.name, self.team,model_name=self.model_name)
diff --git a/flo_ai/router/flo_llm_router.py b/flo_ai/router/flo_llm_router.py
index e5bc0ed0..b8db6d0a 100644
--- a/flo_ai/router/flo_llm_router.py
+++ b/flo_ai/router/flo_llm_router.py
@@ -72,6 +72,7 @@ def __init__(
             flo_team: FloTeam,
             router_prompt: str = None,
             llm: Union[BaseLanguageModel, None] = None,
+            model_nick_name: str = 'default',
         ) -> None:
             self.name = name
             self.session = session
@@ -79,6 +80,7 @@ def __init__(
             self.flo_team = flo_team
             self.agents = flo_team.members
             self.members = [agent.name for agent in flo_team.members]
+            self.model_name = model_nick_name
             self.options = self.members + [FLO_FINISH]
             member_type = (
                 'workers' if flo_team.members[0].type == 'agent' else 'team members'
@@ -118,4 +120,5 @@ def build(self):
                 flo_team=self.flo_team,
                 name=self.name,
                 session=self.session,
+                model_name=self.model_name, 
             )

From 0e7b3f60fd342d386eae2ed39af5c8eda980d22f Mon Sep 17 00:00:00 2001
From: sanosh <sanosh@vizsatizs-MacBook-Pro.local>
Date: Mon, 18 Nov 2024 13:52:18 +0530
Subject: [PATCH 2/3] solving suggestions

---
 flo_ai/router/flo_linear.py     | 17 +++++++++++++----
 flo_ai/router/flo_llm_router.py |  2 +-
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/flo_ai/router/flo_linear.py b/flo_ai/router/flo_linear.py
index cdc8b20e..e890b810 100644
--- a/flo_ai/router/flo_linear.py
+++ b/flo_ai/router/flo_linear.py
@@ -8,7 +8,12 @@
 
 
 class FloLinear(FloRouter):
-    def __init__(self, session: FloSession, name: str, flo_team: FloTeam):
+    def __init__(
+        self,
+        session: FloSession,
+        name: str,
+        flo_team: FloTeam,
+    ):
         super().__init__(
             session=session,
             name=name,
@@ -61,11 +66,15 @@ def create(session: FloSession, name: str, team: FloTeam):
         return FloLinear.Builder(session=session, name=name, flo_team=team).build()
 
     class Builder:
-        def __init__(self, session: FloSession, name: str, flo_team: FloTeam, model_nick_name: str = 'default') -> None:
+        def __init__(
+            self,
+            session: FloSession,
+            name: str,
+            flo_team: FloTeam,
+        ) -> None:
             self.name = name
             self.session = session
             self.team = flo_team
-            self.model_name = model_nick_name
 
         def build(self):
-            return FloLinear(self.session, self.name, self.team,model_name=self.model_name)
+            return FloLinear(self.session, self.name, self.team)
diff --git a/flo_ai/router/flo_llm_router.py b/flo_ai/router/flo_llm_router.py
index b8db6d0a..a98e297a 100644
--- a/flo_ai/router/flo_llm_router.py
+++ b/flo_ai/router/flo_llm_router.py
@@ -120,5 +120,5 @@ def build(self):
                 flo_team=self.flo_team,
                 name=self.name,
                 session=self.session,
-                model_name=self.model_name, 
+                model_name=self.model_name,
             )

From 68552360306d320f7de8a4f6a9fa06b7c3f4653d Mon Sep 17 00:00:00 2001
From: sanosh <sanosh@vizsatizs-MacBook-Pro.local>
Date: Mon, 18 Nov 2024 14:18:09 +0530
Subject: [PATCH 3/3] solving issues

---
 flo_ai/router/flo_llm_router.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/flo_ai/router/flo_llm_router.py b/flo_ai/router/flo_llm_router.py
index a98e297a..38b8a5b1 100644
--- a/flo_ai/router/flo_llm_router.py
+++ b/flo_ai/router/flo_llm_router.py
@@ -24,7 +24,7 @@ def __init__(
         executor: Runnable,
         flo_team: FloTeam,
         name: str,
-        model_name: str,
+        model_name: str = 'default',
     ) -> None:
         super().__init__(
             session=session,
@@ -56,12 +56,14 @@ def create(
         router_prompt: str = None,
         llm: Union[BaseLanguageModel, None] = None,
     ):
+        model_name = 'default' if llm is None else llm.name
         return FloLLMRouter.Builder(
             session=session,
             name=name,
             flo_team=team,
             router_prompt=router_prompt,
             llm=llm,
+            model_nick_name=model_name,
         ).build()
 
     class Builder: