add ids

Josh-XT · Jan 14, 2025 · 5020ee7 · 5020ee7
1 parent d77123c
commit 5020ee7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/agixt/Agent.py b/agixt/Agent.py
@@ -536,6 +536,11 @@ def get_agent_config(self):
                         if value == "":
                             continue
                         config["settings"][key] = value
+        else:
+            company_id = self.auth.company_id
+            self.update_agent_config(
+                new_config={"company_id": company_id}, config_key="settings"
+            )
         return config
 
     async def inference(

diff --git a/agixt/MagicalAuth.py b/agixt/MagicalAuth.py
@@ -303,6 +303,15 @@ def get_agents(email, company=None):
         if company_id and company:
             if company_id != company:
                 continue
+        if not company_id:
+            auth = MagicalAuth(token=impersonate_user(email))
+            company_id = auth.company_id
+            # add to 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,