fix: tool use changing based on if models support tools and agent & fixes for types and stubs in caps

Author: alashchev17

refact-agent/gui/src/__fixtures__/caps.ts

@@ -339,13 +339,15 @@ export const STUB_CAPS_RESPONSE: CapsResponse = {
     "groq-llama-3.1-8b",
     "groq-llama-3.1-70b",
   ],
+  code_chat_default_system_prompt: "default",
   caps_version: 0,
 };
 
 export const EMPTY_CAPS_RESPONSE: CapsResponse = {
   caps_version: 0,
   cloud_name: "",
   code_chat_default_model: "",
+  code_chat_default_system_prompt: "",
   code_chat_models: {},
   code_completion_default_model: "",
   code_completion_models: {},

refact-agent/gui/src/components/ChatForm/PromptSelect.tsx

@@ -32,11 +32,13 @@ export const PromptSelect: React.FC = () => {
   );
 
   const caps = useGetCapsQuery();
+
   const default_system_prompt =
     caps.data?.code_chat_default_system_prompt ?? "default";
+
   const val = useMemo(
     () => Object.keys(selectedSystemPrompt)[0] ?? default_system_prompt,
-    [selectedSystemPrompt],
+    [selectedSystemPrompt, default_system_prompt],
   );
 
   const options = useMemo(() => {
@@ -48,6 +50,12 @@ export const PromptSelect: React.FC = () => {
     });
   }, [promptsRequest.data]);
 
+  const isLoading = useMemo(
+    () =>
+      promptsRequest.isLoading || promptsRequest.isFetching || caps.isLoading,
+    [promptsRequest.isLoading, promptsRequest.isFetching, caps.isLoading],
+  );
+
   return (
     <Flex
       gap="2"
@@ -60,7 +68,7 @@ export const PromptSelect: React.FC = () => {
       <Text size="2" wrap="nowrap">
         System Prompt:
       </Text>
-      <Skeleton loading={promptsRequest.isLoading || promptsRequest.isFetching}>
+      <Skeleton loading={isLoading}>
         <Box flexGrow="1" flexShrink="0">
           <Select
             name="system prompt"

refact-agent/gui/src/hooks/useCapsForToolUse.ts

@@ -8,7 +8,11 @@ import {
   useAppDispatch,
 } from ".";
 
-import { getSelectedChatModel, setChatModel } from "../features/Chat";
+import {
+  getSelectedChatModel,
+  setChatModel,
+  setToolUse,
+} from "../features/Chat";
 
 // TODO: hard coded for now.
 const PAID_AGENT_LIST = [
@@ -48,6 +52,20 @@ export function useCapsForToolUse() {
     return true;
   }, [caps.data?.code_chat_models, currentModel]);
 
+  const modelsSupportingTools = useMemo(() => {
+    const models = caps.data?.code_chat_models ?? {};
+    return Object.entries(models)
+      .filter(([_, value]) => value.supports_tools)
+      .map(([key]) => key);
+  }, [caps.data?.code_chat_models]);
+
+  const modelsSupportingAgent = useMemo(() => {
+    const models = caps.data?.code_chat_models ?? {};
+    return Object.entries(models)
+      .filter(([_, value]) => value.supports_agent)
+      .map(([key]) => key);
+  }, [caps.data?.code_chat_models]);
+
   const usableModels = useMemo(() => {
     const models = caps.data?.code_chat_models ?? {};
     const items = Object.entries(models).reduce<string[]>(
@@ -97,6 +115,22 @@ export function useCapsForToolUse() {
     }
   }, [currentModel, setCapModel, usableModels, usableModelsForPlan]);
 
+  useEffect(() => {
+    if (caps.isSuccess) {
+      if (toolUse === "agent" && modelsSupportingAgent.length === 0) {
+        dispatch(setToolUse("explore"));
+      } else if (toolUse === "explore" && modelsSupportingTools.length === 0) {
+        dispatch(setToolUse("quick"));
+      }
+    }
+  }, [
+    dispatch,
+    caps.isSuccess,
+    toolUse,
+    modelsSupportingAgent,
+    modelsSupportingTools,
+  ]);
+
   return {
     usableModels,
     usableModelsForPlan,

refact-agent/gui/src/services/refact/caps.ts

@@ -79,6 +79,7 @@ export type CapsResponse = {
   caps_version: number;
   cloud_name: string;
   code_chat_default_model: string;
+  code_chat_default_system_prompt: string;
   code_chat_models: Record<string, CodeChatModel>;
   code_completion_default_model: string;
   code_completion_models: Record<string, CodeCompletionModel>;