Skip to content

Commit d85abba

Browse files
committed
[onechat] sanitize tool ids for conv history
1 parent fd39e74 commit d85abba

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

x-pack/platform/packages/shared/onechat/onechat-genai-utils/langchain/index.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ export {
2323
toolsToLangchain,
2424
toolToLangchain,
2525
toolIdentifierFromToolCall,
26+
sanitizeToolId,
2627
type ToolIdMapping,
2728
type ToolsAndMappings,
2829
} from './tools';

x-pack/platform/plugins/shared/onechat/server/services/agents/modes/utils/to_langchain_messages.ts

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,14 @@
55
* 2.0.
66
*/
77

8+
import { BaseMessage, AIMessage, HumanMessage, ToolMessage } from '@langchain/core/messages';
89
import {
910
ConversationRound,
1011
RoundInput,
1112
ToolCallWithResult,
1213
isToolCallStep,
1314
} from '@kbn/onechat-common';
14-
import { BaseMessage, AIMessage, HumanMessage, ToolMessage } from '@langchain/core/messages';
15+
import { sanitizeToolId } from '@kbn/onechat-genai-utils/langchain';
1516

1617
/**
1718
* Converts a conversation to langchain format
@@ -69,12 +70,14 @@ const createAssistantMessage = ({ content }: { content: string }): AIMessage =>
6970
};
7071

7172
export const createToolCallMessages = (toolCall: ToolCallWithResult): [AIMessage, ToolMessage] => {
73+
const toolName = sanitizeToolId(toolCall.tool_id);
74+
7275
const toolCallMessage = new AIMessage({
7376
content: '',
7477
tool_calls: [
7578
{
7679
id: toolCall.tool_call_id,
77-
name: toolCall.tool_id,
80+
name: toolName,
7881
args: toolCall.params,
7982
type: 'tool_call',
8083
},

0 commit comments

Comments
 (0)