From c94505ab39d8855ef7d08bc7c9303725eea6f38a Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Fri, 26 Apr 2024 14:40:00 -0600
Subject: [PATCH] anthropic system prompt (#13131)
---
.../llama_index/llms/anthropic/utils.py | 4 ++--
.../llms/llama-index-llms-anthropic/pyproject.toml | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py b/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py
index 961eabf048..5eb3662b3f 100644
--- a/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py
+++ b/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py
@@ -67,7 +67,7 @@ def messages_to_anthropic_messages(
system_prompt = ""
for message in messages:
if message.role == MessageRole.SYSTEM:
- system_prompt = message.content
+ system_prompt += message.content + "\n"
elif message.role == MessageRole.FUNCTION or message.role == MessageRole.TOOL:
content = ToolResultBlockParam(
tool_use_id=message.additional_kwargs["tool_call_id"],
@@ -105,7 +105,7 @@ def messages_to_anthropic_messages(
)
anthropic_messages.append(anth_message)
- return __merge_common_role_msgs(anthropic_messages), system_prompt
+ return __merge_common_role_msgs(anthropic_messages), system_prompt.strip()
# Function used in bedrock
diff --git a/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml
index 67cb500377..64cd8a2627 100644
--- a/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-llms-anthropic"
readme = "README.md"
-version = "0.1.10"
+version = "0.1.11"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
--
GitLab