set company id if not set

Josh-XT · Jan 13, 2025 · b126f7b · b126f7b
1 parent b4bb122
commit b126f7b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/agixt/Agent.py b/agixt/Agent.py
@@ -232,6 +232,17 @@ def get_agents(user=DEFAULT_USER, company=None):
         if company_id and company:
             if company_id != company:
                 continue
+        if not company_id:
+            auth = MagicalAuth(token=impersonate_user(user_id=str(user_data.id)))
+            company_id = str(auth.company_id)
+            # update agent settings
+            agent_setting = AgentSettingModel(
+                agent_id=agent.id,
+                name="company_id",
+                value=company_id,
+            )
+            session.add(agent_setting)
+            session.commit()
         output.append(
             {
                 "name": agent.name,