mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-09-01 23:37:34 +00:00
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:
@@ -21,8 +21,6 @@ from llama_index.llms.llama_utils import (
|
||||
)
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from collections.abc import Callable
|
||||
|
||||
from llama_index.callbacks import CallbackManager
|
||||
from llama_index.llms import (
|
||||
CompletionResponseGen,
|
||||
@@ -113,10 +111,10 @@ class SagemakerLLM(CustomLLM):
|
||||
context_window: int = Field(
|
||||
description="The maximum number of context tokens for the model."
|
||||
)
|
||||
messages_to_prompt: Callable[..., str] = Field(
|
||||
messages_to_prompt: Any = Field(
|
||||
description="The function to convert messages to a prompt.", exclude=True
|
||||
)
|
||||
completion_to_prompt: Callable[..., str] = Field(
|
||||
completion_to_prompt: Any = Field(
|
||||
description="The function to convert a completion to a prompt.", exclude=True
|
||||
)
|
||||
generate_kwargs: dict[str, Any] = Field(
|
||||
|
Reference in New Issue
Block a user