Merge pull request #221 from bigadsoleiman/main

Add Llama2 support for Amazon Bedrock. Fix prompts for Llama2 on SageMaker.

lib/chatbot-api/functions/api-handler/routes/embeddings.py

@@ -37,7 +37,6 @@ def embeddings():
     if selected_model is None:
         raise genai_core.types.CommonError("Model not found")
 
-    ret_value = genai_core.embeddings.generate_embeddings(
-        selected_model, request.input)
+    ret_value = genai_core.embeddings.generate_embeddings(selected_model, request.input)
 
     return {"ok": True, "data": ret_value}

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

@@ -2,3 +2,4 @@
 from .sagemaker import *
 from .bedrock import *
 from .base import Mode
+from .shared import *

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

@@ -73,7 +73,7 @@ def get_prompt(self):
         {chat_history}
 
         Question: {input}"""
-        
+
         return PromptTemplate.from_template(template)
 
     def get_condense_question_prompt(self):

lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/__init__.py

@@ -2,3 +2,4 @@
 from .titan import *
 from .ai21_j2 import *
 from .cohere import *
+from .llama2_chat import *