fix: Docker and sagemaker setup (#1118)

* fix: docker copying extra files

* feat: allow configuring mode through env vars

* feat: Attempt to build and tag a docker image

* fix: run docker on release

* fix: typing in prompt transformation

* chore: remove tutorial comments
This commit is contained in:
Pablo Orgaz
2023-10-27 13:29:29 +02:00
committed by GitHub
parent 768e5ff505
commit 895588b82a
7 changed files with 67 additions and 15 deletions

View File

@@ -37,6 +37,8 @@ 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