Skip to content

Commit 33b3931

Browse files
massi-angbigadsoleiman
authored andcommitted
feat(multimodal): fixes to initialize the adapters
1 parent 0be52ea commit 33b3931

File tree

8 files changed

+10
-9
lines changed

8 files changed

+10
-9
lines changed

lib/model-interfaces/langchain/functions/request-handler/adapters/openai/gpt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
from langchain.chat_models import ChatOpenAI
33
from ..base import ModelAdapter
4-
from ..registry import registry
4+
from genai_core.registry import registry
55

66

77
class GPTAdapter(ModelAdapter):

lib/model-interfaces/langchain/functions/request-handler/adapters/sagemaker/amazon/falconlite.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from langchain.prompts.prompt import PromptTemplate
66

77
from ...base import ModelAdapter
8-
from ...registry import registry
8+
from genai_core.registry import registry
99

1010

1111
class FalconLiteContentHandler(LLMContentHandler):

lib/model-interfaces/langchain/functions/request-handler/adapters/sagemaker/meta/llama2_chat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from langchain.llms.sagemaker_endpoint import LLMContentHandler, SagemakerEndpoint
55

66
from ...base import ModelAdapter
7-
from ...registry import registry
7+
from genai_core.registry import registry
88

99
from ...shared.meta.llama2_chat import (
1010
Llama2ChatPromptTemplate,

lib/model-interfaces/langchain/functions/request-handler/adapters/sagemaker/mistralai/mistral_instruct.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from langchain.prompts.prompt import PromptTemplate
66

77
from ...base import ModelAdapter
8-
from ...registry import registry
8+
from genai_core.registry import registry
99

1010

1111
class MistralInstructContentHandler(LLMContentHandler):

lib/model-interfaces/langchain/functions/request-handler/adapters/sagemaker/mistralai/mixtral_instruct.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from langchain.prompts.prompt import PromptTemplate
66

77
from ...base import ModelAdapter
8-
from ...registry import registry
8+
from genai_core.registry import registry
99

1010

1111
class MixtralInstructContentHandler(LLMContentHandler):

lib/model-interfaces/langchain/functions/request-handler/index.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
1111
from aws_lambda_powertools.utilities.typing import LambdaContext
1212

13+
import adapters
1314
from genai_core.utils.websocket import send_to_client
1415
from genai_core.types import ChatbotAction
1516

lib/user-interface/react-app/src/components/chatbot/chat-input-panel.tsx

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -328,9 +328,10 @@ export default function ChatInputPanel(props: ChatInputPanelProps) {
328328
const value = state.value.trim();
329329
const request: ChatBotRunRequest = {
330330
action: ChatBotAction.Run,
331-
modelInterface: props.configuration.files
332-
? "multimodal"
333-
: (state.selectedModelMetadata!.interface as ModelInterface),
331+
modelInterface:
332+
props.configuration.files && props.configuration.files.length > 0
333+
? "multimodal"
334+
: (state.selectedModelMetadata!.interface as ModelInterface),
334335
data: {
335336
mode: ChatBotMode.Chain,
336337
text: value,

lib/user-interface/react-app/src/components/chatbot/chat-message.tsx

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ import {
44
Container,
55
ExpandableSection,
66
Popover,
7-
SpaceBetween,
87
Spinner,
98
StatusIndicator,
109
Tabs,

0 commit comments

Comments
 (0)