Merge branch 'main' into stas/retry-grant

Author: smoreinis

.vscode/settings.json

@@ -12,7 +12,7 @@
     "ruff.fixAll": true,
     "python.defaultInterpreterPath": "${workspaceFolder}/.venv/bin/python",
     "python.terminal.activateEnvironment": true,
-    "python.languageServer": "Pylance",
+    "python.languageServer": "None",
     "editor.defaultFormatter": "esbenp.prettier-vscode",
     "editor.formatOnSave": true,
     "editor.codeActionsOnSave": {

agentex-ui/app/page.tsx

@@ -1,6 +1,7 @@
 import { connection } from 'next/server';
 
 import { AgentexUIRoot } from '@/components/agentex-ui-root';
+import { AgentexProvider } from '@/components/providers';
 
 export default async function RootPage() {
   await connection();
@@ -19,9 +20,11 @@ export default async function RootPage() {
   }
 
   return (
-    <AgentexUIRoot
+    <AgentexProvider
       sgpAppURL={sgpAppURL}
       agentexAPIBaseURL={agentexAPIBaseURL}
-    />
+    >
+      <AgentexUIRoot />
+    </AgentexProvider>
   );
 }

agentex-ui/components/agentex-ui-root.tsx

@@ -5,36 +5,41 @@ import { useCallback, useEffect, useState } from 'react';
 import { ToastContainer } from 'react-toastify';
 
 import { PrimaryContent } from '@/components/primary-content/primary-content';
-import { AgentexProvider } from '@/components/providers';
+import { useAgentexClient } from '@/components/providers';
 import { TaskSidebar } from '@/components/task-sidebar/task-sidebar';
 import { TracesSidebar } from '@/components/traces-sidebar/traces-sidebar';
+import { useAgents } from '@/hooks/use-agents';
 import { useLocalStorageState } from '@/hooks/use-local-storage-state';
 import {
   SearchParamKey,
   useSafeSearchParams,
 } from '@/hooks/use-safe-search-params';
 
-type AgentexUIRootProps = {
-  sgpAppURL: string;
-  agentexAPIBaseURL: string;
-};
-
-export function AgentexUIRoot({
-  sgpAppURL,
-  agentexAPIBaseURL,
-}: AgentexUIRootProps) {
+export function AgentexUIRoot() {
   const { agentName, taskID, updateParams } = useSafeSearchParams();
   const [isTracesSidebarOpen, setIsTracesSidebarOpen] = useState(false);
+  const { agentexClient } = useAgentexClient();
+  const { data: agents = [], isLoading } = useAgents(agentexClient);
   const [localAgentName, setLocalAgentName] = useLocalStorageState<
     string | undefined
   >('lastSelectedAgent', undefined);
 
   useEffect(() => {
+    if (isLoading) return;
+
+    const selectedAgent = agents.find(agent => agent.name === agentName);
+    const isAgentValid = selectedAgent && selectedAgent.status === 'Ready';
+
+    if (!isAgentValid) {
+      updateParams({ [SearchParamKey.AGENT_NAME]: null });
+      setLocalAgentName(undefined);
+    }
+
     if (!agentName && localAgentName) {
       updateParams({ [SearchParamKey.AGENT_NAME]: localAgentName });
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, []);
+  }, [isLoading]);
 
   const handleSelectTask = useCallback(
     (taskId: string | null) => {
@@ -70,21 +75,16 @@ export function AgentexUIRoot({
 
   return (
     <>
-      <AgentexProvider
-        sgpAppURL={sgpAppURL ?? ''}
-        agentexAPIBaseURL={agentexAPIBaseURL}
-      >
-        <div className="fixed inset-0 flex w-full">
-          <TaskSidebar />
-          <PrimaryContent
-            isTracesSidebarOpen={isTracesSidebarOpen}
-            toggleTracesSidebar={() =>
-              setIsTracesSidebarOpen(!isTracesSidebarOpen)
-            }
-          />
-          <TracesSidebar isOpen={isTracesSidebarOpen} />
-        </div>
-      </AgentexProvider>
+      <div className="fixed inset-0 flex w-full">
+        <TaskSidebar />
+        <PrimaryContent
+          isTracesSidebarOpen={isTracesSidebarOpen}
+          toggleTracesSidebar={() =>
+            setIsTracesSidebarOpen(!isTracesSidebarOpen)
+          }
+        />
+        <TracesSidebar isOpen={isTracesSidebarOpen} />
+      </div>
       <ToastContainer />
     </>
   );

agentex-ui/components/primary-content/chat-view.tsx

@@ -63,7 +63,7 @@ export function ChatView({
         ref={headerRef}
       />
 
-      <div className="flex min-h-full w-full flex-col items-center px-4 sm:px-6 md:px-8">
+      <div className="flex w-full flex-col items-center px-4 sm:px-6 md:px-8">
         <div className="w-full max-w-3xl">
           <TaskProvider taskId={taskID}>
             <TaskMessages taskId={taskID} headerRef={headerRef} />

agentex-ui/components/task-header/task-header.tsx

@@ -80,7 +80,11 @@ export function TaskHeader({
               </SelectTrigger>
               <SelectContent>
                 {agents.map(agent => (
-                  <SelectItem key={agent.name} value={agent.name}>
+                  <SelectItem
+                    key={agent.name}
+                    value={agent.name}
+                    disabled={agent.status !== 'Ready'}
+                  >
                     {agent.name}
                   </SelectItem>
                 ))}

agentex-ui/components/task-messages/task-message-reasoning-content.tsx

@@ -1,4 +1,4 @@
-import { memo, useState, useMemo } from 'react';
+import { memo, useState, useMemo, useRef, useEffect } from 'react';
 
 import { motion } from 'framer-motion';
 import { BrainIcon, ChevronDownIcon } from 'lucide-react';
@@ -21,6 +21,9 @@ type TaskMessageReasoningProps = {
 
 function TaskMessageReasoningImpl({ message }: TaskMessageReasoningProps) {
   const [isCollapsed, setIsCollapsed] = useState(true);
+  const [showTopBlur, setShowTopBlur] = useState(false);
+  const [showBottomBlur, setShowBottomBlur] = useState(false);
+  const scrollContainerRef = useRef<HTMLDivElement>(null);
 
   const { taskID } = useSafeSearchParams();
   const { agentexClient } = useAgentexClient();
@@ -61,6 +64,30 @@ function TaskMessageReasoningImpl({ message }: TaskMessageReasoningProps) {
     ].join('\n\n');
   }, [message.content]);
 
+  const updateBlurEffects = () => {
+    const container = scrollContainerRef.current;
+    if (!container) return;
+
+    const { scrollTop, scrollHeight, clientHeight } = container;
+    const isScrollable = scrollHeight > clientHeight;
+
+    setShowTopBlur(isScrollable && scrollTop > 10);
+    setShowBottomBlur(
+      isScrollable && scrollTop < scrollHeight - clientHeight - 10
+    );
+  };
+
+  useEffect(() => {
+    const container = scrollContainerRef.current;
+    if (!container) return;
+
+    updateBlurEffects();
+    const resizeObserver = new ResizeObserver(updateBlurEffects);
+    resizeObserver.observe(container);
+
+    return () => resizeObserver.disconnect();
+  }, [isCollapsed, reasoningText]);
+
   return (
     <motion.div
       className="w-full"
@@ -90,9 +117,29 @@ function TaskMessageReasoningImpl({ message }: TaskMessageReasoningProps) {
         />
       </button>
       <Collapsible collapsed={isCollapsed}>
-        <MarkdownResponse className="ml-6 grid border-l-4 border-gray-300 pl-3 text-gray-500">
-          {reasoningText}
-        </MarkdownResponse>
+        <div className="relative ml-6">
+          <div
+            className={cn(
+              'pointer-events-none absolute inset-x-0 top-0 z-10 h-8 bg-gradient-to-b from-white to-transparent transition-opacity',
+              showTopBlur ? 'opacity-100' : 'opacity-0'
+            )}
+          />
+          <div
+            ref={scrollContainerRef}
+            onScroll={updateBlurEffects}
+            className="max-h-48 overflow-y-auto"
+          >
+            <MarkdownResponse className="grid border-l-4 border-gray-300 pl-3 text-gray-500">
+              {reasoningText}
+            </MarkdownResponse>
+          </div>
+          <div
+            className={cn(
+              'pointer-events-none absolute inset-x-0 bottom-0 z-10 h-8 bg-gradient-to-t from-white to-transparent transition-opacity',
+              showBottomBlur ? 'opacity-100' : 'opacity-0'
+            )}
+          />
+        </div>
       </Collapsible>
     </motion.div>
   );

agentex-ui/package-lock.json

@@ -0,0 +1,33 @@
+# Enterprise Edition
+
+## Overview
+
+Agentex Enterprise Edition offers enhanced capabilities to support, monitor, and evaluate Agentex agents. The video below gives a brief demo of the what the developer experience would look like with Agentex's enterprise features enabled 
+
+
+<video controls width="100%" >
+  <source src="https://github.com/user-attachments/assets/cccb6211-281c-49d5-b24c-6730ae576b14" type="video/mp4">
+</video>
+
+## Why Open Source?
+
+At Scale, we've spent the last three years building enterprise AI agents and learned how different every use case is. To unify our approach, we built a single delivery framework and now we're open-sourcing it to share what we've learned. Many enterprises have built upon open source tooling, and we want to contribute to that ecosystem. Our goal is simple: see more useful AI in production.
+
+Agentex is also cloud-agnostic and Kubernetes-native. We intentionally kept it lightweight and unopinionated to maximize flexibility and to incur minimal infrastructure and security overhead.
+
+Here are the differences between Open Source vs Enterprise to meet different organizational needs:
+
+| Feature | Open Source Edition | Enterprise Edition |
+|---------|--------------------|--------------------|
+| **Source Code** | ✅ Open source server, developer UI, and SDK | ✅ Open source server, developer UI, and SDK |
+| **Local Development** | ✅ Use this repo for local development | ✅ Use this repo for local development |
+| **Community Support** | ✅ GitHub issues, discussions, pull requests | ✅ GitHub issues, discussions, pull requests |
+| **GitOps Setup** | ❌ DIY deployment using public helm charts and the `agentex` CLI in CI/CD | ✅ Scale sets up CI/CD on select repositories for automatic agent deployment |
+| **Builder Tools** | ❌ Bring your own (vector stores, models, etc.) | ✅ Model inference, knowledge bases, etc. |
+| **Agent Operations (AgentOps)** | ❌ Not included | ✅ Full agent lifecycle management: hosting, version control, interaction UI, tracing, evaluation |
+| **Identity Management** | ❌ No user management | ✅ SSO/SAML authentication, centralized API key management |
+| **Enterprise Operations** | ❌ Self-service setup | ✅ Uptime/availability SLAs, security reviews, deployment, installation, ongoing maintenance |
+
+**Ready for Enterprise?** Contact our team at https://scale.com/demo to discuss your requirements.
+
+> For our current and future customers, Agentex is a module that is hosted and deployed as part of the Scale GenAI Platform's Enterprise License. This open source project is meant to give people a local development ability and community support.

agentex/docs/docs/development_guides/enterprise_edition.md

@@ -75,6 +75,7 @@ nav:
       - State Machines: development_guides/state_machines.md
       - Agent Task Tracker: development_guides/agent_task_tracker.md
       - Webhooks: development_guides/webhooks.md
+      - Enterprise Edition: development_guides/enterprise_edition.md
   - Temporal Development:
       - Overview: temporal_development/overview.md
       - Use Cases: temporal_development/use_cases.md