Remove tool calling for Cohere v1

559f1b81f79b2c4e2dc68538d29ead31ed738c36

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: 01f03dbf508a00240e3019c517d8f61166af9eb1

public/scripts/tool-calling.js+0 -13 Viewed

+        }
         */
-        /*
-        if ([chat_completion_sources.COHERE].includes(oai_settings.chat_completion_source)) {
-            if (!Array.isArray(data?.tool_calls)) {
-                return;
-            }
-            for (const toolCall of data.tool_calls) {
-                const args = { name: toolCall.name, arguments: JSON.stringify(toolCall.parameters) };
-                console.log('Function tool call:', toolCall);
-            }
-        }
-        */
         return result;
+    }

src/endpoints/backends/chat-completions.js+1 -9 Viewed

 const { jsonParser } = require('../../express-common');
 const { CHAT_COMPLETION_SOURCES, GEMINI_SAFETY, BISON_SAFETY, OPENROUTER_HEADERS } = require('../../constants');
 const { forwardFetchResponse, getConfigValue, tryParse, uuidv4, mergeObjectWithYaml, excludeKeysByYaml, color } = require('../../util');
-const { convertClaudeMessages, convertGooglePrompt, convertTextCompletionPrompt, convertCohereMessages, convertMistralMessages, convertCohereTools, convertAI21Messages } = require('../../prompt-converters');
+const { convertClaudeMessages, convertGooglePrompt, convertTextCompletionPrompt, convertCohereMessages, convertMistralMessages, convertAI21Messages } = require('../../prompt-converters');
 const CohereStream = require('../../cohere-stream');
 const { readSecret, SECRET_KEYS } = require('../secrets');

             });
+        }
-        /*
-        if (Array.isArray(request.body.tools) && request.body.tools.length > 0) {
-            tools.push(...convertCohereTools(request.body.tools));
-            // Can't have both connectors and tools in the same request
-            connectors.splice(0, connectors.length);
-        }
-        */
         // https://docs.cohere.com/reference/chat
         const requestBody = {
             stream: Boolean(request.body.stream),

src/prompt-converters.js+0 -71 Viewed

     return messageStrings.join('\n') + '\nassistant:';
+}
-/**
- * Convert OpenAI Chat Completion tools to the format used by Cohere.
- * @param {object[]} tools OpenAI Chat Completion tool definitions
- */
-function convertCohereTools(tools) {
-    if (!Array.isArray(tools) || tools.length === 0) {
-        return [];
-    }
-    const jsonSchemaToPythonTypes = {
-        'string': 'str',
-        'number': 'float',
-        'integer': 'int',
-        'boolean': 'bool',
-        'array': 'list',
-        'object': 'dict',
-    };
-    const cohereTools = [];
-    for (const tool of tools) {
-        if (tool?.type !== 'function') {
-            console.log(`Unsupported tool type: ${tool.type}`);
-            continue;
-        }
-        const name = tool?.function?.name;
-        const description = tool?.function?.description;
-        const properties = tool?.function?.parameters?.properties;
-        const required = tool?.function?.parameters?.required;
-        const parameters = {};
-        if (!name) {
-            console.log('Tool name is missing');
-            continue;
-        }
-        if (!description) {
-            console.log('Tool description is missing');
-        }
-        if (!properties || typeof properties !== 'object') {
-            console.log(`No properties found for tool: ${tool?.function?.name}`);
-            continue;
-        }
-        for (const property in properties) {
-            const parameterDefinition = properties[property];
-            const description = parameterDefinition.description || (parameterDefinition.enum ? JSON.stringify(parameterDefinition.enum) : '');
-            const type = jsonSchemaToPythonTypes[parameterDefinition.type] || 'str';
-            const isRequired = Array.isArray(required) && required.includes(property);
-            parameters[property] = {
-                description: description,
-                type: type,
-                required: isRequired,
-            };
-        }
-        const cohereTool = {
-            name: tool.function.name,
-            description: tool.function.description,
-            parameter_definitions: parameters,
-        };
-        cohereTools.push(cohereTool);
-    }
-    return cohereTools;
-}
 module.exports = {
     convertClaudePrompt,
     convertClaudeMessages,

     convertTextCompletionPrompt,
     convertCohereMessages,
     convertMistralMessages,
-    convertCohereTools,
     convertAI21Messages,
 };