stuff agents into types and update default ports

Author: maxkorp

typescript-sdk/apps/dojo/src/agents.ts

@@ -12,9 +12,11 @@ import { LangGraphAgent, LangGraphHttpAgent } from "@ag-ui/langgraph";
 import { AgnoAgent } from "@ag-ui/agno";
 import { LlamaIndexAgent } from "@ag-ui/llamaindex";
 import { CrewAIAgent } from "@ag-ui/crewai";
+import getEnvVars from "./env";
 import { mastra } from "./mastra";
 import { PydanticAIAgent } from "@ag-ui/pydantic-ai";
 
+const envVars = getEnvVars();
 export const agentsIntegrations: AgentIntegrationConfig[] = [
   {
     id: "middleware-starter",
@@ -53,7 +55,7 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     id: "server-starter",
     agents: async () => {
       return {
-        agentic_chat: new ServerStarterAgent({ url: "http://localhost:8000/" }),
+        agentic_chat: new ServerStarterAgent({ url: envVars.serverStarterUrl }),
       };
     },
   },
@@ -62,22 +64,22 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     agents: async () => {
       return {
         agentic_chat: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/agentic_chat",
+          url: `${envVars.serverStarterAllFeaturesUrl}/agentic_chat`,
         }),
         human_in_the_loop: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/human_in_the_loop",
+          url: `${envVars.serverStarterAllFeaturesUrl}/human_in_the_loop`,
         }),
         agentic_generative_ui: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/agentic_generative_ui",
+          url: `${envVars.serverStarterAllFeaturesUrl}/agentic_generative_ui`,
         }),
         tool_based_generative_ui: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/tool_based_generative_ui",
+          url: `${envVars.serverStarterAllFeaturesUrl}/tool_based_generative_ui`,
         }),
         shared_state: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/shared_state",
+          url: `${envVars.serverStarterAllFeaturesUrl}/shared_state`,
         }),
         predictive_state_updates: new ServerStarterAllFeaturesAgent({
-          url: "http://localhost:8000/predictive_state_updates",
+          url: `${envVars.serverStarterAllFeaturesUrl}/predictive_state_updates`,
         }),
       };
     },
@@ -86,7 +88,7 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     id: "mastra",
     agents: async () => {
       const mastraClient = new MastraClient({
-        baseUrl: "http://localhost:4111",
+        baseUrl: envVars.mastraUrl,
       });
 
       return MastraAgent.getRemoteAgents({
@@ -113,27 +115,27 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     agents: async () => {
       return {
         agentic_chat: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "agentic_chat",
         }),
         agentic_generative_ui: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "agentic_generative_ui",
         }),
         human_in_the_loop: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "human_in_the_loop",
         }),
         predictive_state_updates: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "predictive_state_updates",
         }),
         shared_state: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "shared_state",
         }),
         tool_based_generative_ui: new LangGraphAgent({
-          deploymentUrl: "http://localhost:2024",
+          deploymentUrl: envVars.langgraphUrl,
           graphId: "tool_based_generative_ui",
         }),
       };
@@ -169,7 +171,7 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     agents: async () => {
       return {
         agentic_chat: new AgnoAgent({
-          url: "http://localhost:8000/agui",
+          url: `${envVars.agnoUrl}/agui`,
         }),
       };
     },
@@ -179,16 +181,16 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     agents: async () => {
       return {
         agentic_chat: new LlamaIndexAgent({
-          url: "http://localhost:9000/agentic_chat/run",
+          url: `${envVars.llamaIndexUrl}/agentic_chat/run`,
         }),
         human_in_the_loop: new LlamaIndexAgent({
-          url: "http://localhost:9000/human_in_the_loop/run",
+          url: `${envVars.llamaIndexUrl}/human_in_the_loop/run`,
         }),
         agentic_generative_ui: new LlamaIndexAgent({
-          url: "http://localhost:9000/agentic_generative_ui/run",
+          url: `${envVars.llamaIndexUrl}/agentic_generative_ui/run`,
         }),
         shared_state: new LlamaIndexAgent({
-          url: "http://localhost:9000/shared_state/run",
+          url: `${envVars.llamaIndexUrl}/shared_state/run`,
         }),
       };
     },
@@ -198,22 +200,22 @@ export const agentsIntegrations: AgentIntegrationConfig[] = [
     agents: async () => {
       return {
         agentic_chat: new CrewAIAgent({
-          url: "http://localhost:8000/agentic_chat",
+          url: `${envVars.crewAiUrl}/agentic_chat`,
         }),
         human_in_the_loop: new CrewAIAgent({
-          url: "http://localhost:8000/human_in_the_loop",
+          url: `${envVars.crewAiUrl}/human_in_the_loop`,
         }),
         tool_based_generative_ui: new CrewAIAgent({
-          url: "http://localhost:8000/tool_based_generative_ui",
+          url: `${envVars.crewAiUrl}/tool_based_generative_ui`,
         }),
         agentic_generative_ui: new CrewAIAgent({
-          url: "http://localhost:8000/agentic_generative_ui",
+          url: `${envVars.crewAiUrl}/agentic_generative_ui`,
         }),
         shared_state: new CrewAIAgent({
-          url: "http://localhost:8000/shared_state",
+          url: `${envVars.crewAiUrl}/shared_state`,
         }),
         predictive_state_updates: new CrewAIAgent({
-          url: "http://localhost:8000/predictive_state_updates",
+          url: `${envVars.crewAiUrl}/predictive_state_updates`,
         }),
       };
     },

typescript-sdk/apps/dojo/src/env.ts

@@ -0,0 +1,21 @@
+type envVars = {
+  serverStarterUrl: string;
+  serverStarterAllFeaturesUrl: string;
+  mastraUrl: string;
+  langgraphUrl: string;   
+  agnoUrl: string;
+  llamaIndexUrl: string;
+  crewAiUrl: string;
+}
+
+export default function getEnvVars(): envVars {
+    return {
+        serverStarterUrl: process.env.SERVER_STARTER_URL || 'http://localhost:8000',
+        serverStarterAllFeaturesUrl: process.env.SERVER_STARTER_ALL_FEATURES_URL || 'http://localhost:8000',
+        mastraUrl: process.env.MASTRA_URL || 'http://localhost:4111',
+        langgraphUrl: process.env.LANGGRAPH_URL || 'http://localhost:2024',
+        agnoUrl: process.env.AGNO_URL || 'http://localhost:9001',
+        llamaIndexUrl: process.env.LLAMA_INDEX_URL || 'http://localhost:9000',
+        crewAiUrl: process.env.CREW_AI_URL || 'http://localhost:9002',
+    }
+}