fix: sagemaker config and chat methods (#1142)

2025-10-14 02:58:03 +00:00 · 2023-10-30 21:54:41 +01:00
parent b0e258265f
commit a517a588c4
2 changed files with 29 additions and 4 deletions
--- a/private_gpt/components/llm/llm_component.py
+++ b/private_gpt/components/llm/llm_component.py
@@ -37,8 +37,6 @@ class LLMComponent:

                self.llm = SagemakerLLM(
                    endpoint_name=settings.sagemaker.endpoint_name,
-                    messages_to_prompt=messages_to_prompt,
-                    completion_to_prompt=completion_to_prompt,
                )
            case "openai":
                from llama_index.llms import OpenAI