Merge pull request #71 from ttt246/feature/16_merge_repos

Brain/src/common/assembler.py

@@ -1,12 +1,12 @@
 # assembler to mapping data into another data type.
 from typing import Any, List
 
-from src.common.http_response_codes import responses
-from src.model.basic_model import BasicModel
-from src.model.contact_model import ContactModel
-from src.model.message_model import MessageModel
-from src.model.requests.request_model import ChatRising, SendSMS, TrainContacts
-from src.model.sms_model import SMSModel
+from Brain.src.common.http_response_codes import responses
+from Brain.src.model.basic_model import BasicModel
+from Brain.src.model.contact_model import ContactModel
+from Brain.src.model.message_model import MessageModel
+from Brain.src.model.requests.request_model import ChatRising, SendSMS, TrainContacts
+from Brain.src.model.sms_model import SMSModel
 
 
 class Assembler:

Brain/src/common/brain_exception.py

@@ -1,7 +1,7 @@
 """Basic Exception in Brain"""
 from typing import Any
 
-from src.common.http_response_codes import responses
+from Brain.src.common.http_response_codes import responses
 
 
 class BrainException(Exception):

Brain/src/common/utils.py

@@ -41,8 +41,8 @@ HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 
 def get_firebase_cred():
-    if os.path.exists("firebase_cred.json"):
-        file = open("firebase_cred.json")
+    if os.path.exists("Brain/firebase_cred.json"):
+        file = open("Brain/firebase_cred.json")
         cred = json.load(file)
         file.close()
         return credentials.Certificate(cred)

Brain/src/firebase/firebase.py

@@ -1,5 +1,5 @@
 import firebase_admin
-from src.common.utils import get_firebase_cred, FIREBASE_STORAGE_BUCKET
+from Brain.src.common.utils import get_firebase_cred, FIREBASE_STORAGE_BUCKET
 
 
 def initialize_app():

Brain/src/logs.py

@@ -9,8 +9,8 @@ from typing import Any
 
 from colorama import Fore, Style
 
-from src.log_cycle.json_handler import JsonFormatter, JsonFileHandler
-from src.singletone import Singleton
+from Brain.src.log_cycle.json_handler import JsonFormatter, JsonFileHandler
+from Brain.src.singletone import Singleton
 
 
 class Logger(metaclass=Singleton):

Brain/src/model/chat_response_model.py

@@ -1,7 +1,7 @@
 """chat response model"""
 from typing import Any
 
-from src.model.message_model import MessageModel
+from Brain.src.model.message_model import MessageModel
 
 
 class ChatResponseModel:

Brain/src/model/contact_model.py

@@ -7,7 +7,7 @@
 }"""
 from typing import Any
 
-from src.model.requests.request_model import TrainContacts
+from Brain.src.model.requests.request_model import TrainContacts
 
 
 class ContactModel:

Brain/src/model/feedback_model.py

@@ -2,7 +2,7 @@ import datetime
 
 
 # feedback model to manage user's feedback (thumb up and down)
-from src.model.basic_model import BasicModel
+from Brain.src.model.basic_model import BasicModel
 
 
 class FeedbackModel:

Brain/src/model/image_model.py

@@ -1,7 +1,7 @@
 """Image model to process & handle them"""
 from typing import Any
 
-from src.model.basic_model import DataStatus
+from Brain.src.model.basic_model import DataStatus
 
 
 class ImageModel:

Brain/src/model/message_model.py

@@ -1,5 +1,5 @@
 """message model to manage browsering questions as agent"""
-from src.service.llm.base import Message
+from Brain.src.service.llm.base import Message
 
 
 class MessageModel:

Brain/src/model/sms_model.py

@@ -1,7 +1,7 @@
 """sms message model includes from, to and body"""
 from typing import Any
 
-from src.model.requests.request_model import SendSMS
+from Brain.src.model.requests.request_model import SendSMS
 
 
 class SMSModel:

Brain/src/rising_plugin/guardrails-config/actions/actions.py

@@ -22,20 +22,20 @@ from langchain.vectorstores import utils
 from langchain.document_loaders.csv_loader import CSVLoader
 from langchain.docstore.document import Document
 
-from src.common.utils import (
+from Brain.src.common.utils import (
     OPENAI_API_KEY,
     COMMAND_SMS_INDEXS,
     COMMAND_BROWSER_OPEN,
 )
-from src.rising_plugin.image_embedding import (
+from Brain.src.rising_plugin.image_embedding import (
     query_image_text,
 )
 
 from nemoguardrails.actions import action
 
-from src.rising_plugin.llm.falcon_llm import FalconLLM
-from src.rising_plugin.llm.gpt_llm import GptLLM
-from src.rising_plugin.llm.llms import (
+from Brain.src.rising_plugin.llm.falcon_llm import FalconLLM
+from Brain.src.rising_plugin.llm.gpt_llm import GptLLM
+from Brain.src.rising_plugin.llm.llms import (
     get_llm_chain,
     GPT_3_5_TURBO,
     GPT_4_32K,

Brain/src/rising_plugin/llm/falcon_llm.py

@@ -1,7 +1,7 @@
 """falcon llm"""
 from langchain import HuggingFaceHub, PromptTemplate, LLMChain
 
-from src.common.utils import HUGGINGFACEHUB_API_TOKEN
+from Brain.src.common.utils import HUGGINGFACEHUB_API_TOKEN
 
 repo_id = "tiiuae/falcon-7b-instruct"
 template = """

Brain/src/rising_plugin/llm/gpt_llm.py

@@ -3,7 +3,7 @@ from typing import Any
 
 from langchain.chat_models import ChatOpenAI
 from langchain.chains.question_answering import load_qa_chain
-from src.common.utils import (
+from Brain.src.common.utils import (
     OPENAI_API_KEY,
 )
 

Brain/src/rising_plugin/llm/llms.py

@@ -1,9 +1,9 @@
 """lLMs which we offer"""
 from typing import Any
 
-from src.common.brain_exception import BrainException
-from src.rising_plugin.llm.falcon_llm import FalconLLM
-from src.rising_plugin.llm.gpt_llm import GptLLM
+from Brain.src.common.brain_exception import BrainException
+from Brain.src.rising_plugin.llm.falcon_llm import FalconLLM
+from Brain.src.rising_plugin.llm.gpt_llm import GptLLM
 
 GPT_3_5_TURBO = "gpt-3.5-turbo"
 GPT_4 = "gpt-4"

Brain/src/router/api.py

@@ -1,11 +1,11 @@
 import json
 import os
 
-from src.common.assembler import Assembler
-from src.common.brain_exception import BrainException
-from src.common.utils import ProgramType
-from src.model.image_model import ImageModel
-from src.model.requests.request_model import (
+from Brain.src.common.assembler import Assembler
+from Brain.src.common.brain_exception import BrainException
+from Brain.src.common.utils import ProgramType
+from Brain.src.model.image_model import ImageModel
+from Brain.src.model.requests.request_model import (
     Notification,
     UploadImage,
     ImageRelatedness,
@@ -17,24 +17,24 @@ from src.model.requests.request_model import (
     ClientInfo,
     get_client_info,
 )
-from src.rising_plugin.risingplugin import (
+from Brain.src.rising_plugin.risingplugin import (
     getCompletion,
     getTextFromImage,
     query_image_ask,
     handle_chat_completion,
 )
-from src.firebase.cloudmessage import send_message, get_tokens
-from src.rising_plugin.csv_embed import csv_embed
-from src.rising_plugin.image_embedding import embed_image_text, query_image_text
-
-from src.logs import logger
-from src.model.basic_model import BasicModel
-from src.model.feedback_model import FeedbackModel
-from src.service.command_service import CommandService
-from src.service.contact_service import ContactsService
-from src.service.feedback_service import FeedbackService
-from src.service.llm.chat_service import ChatService
-from src.service.twilio_service import TwilioService
+from Brain.src.firebase.cloudmessage import send_message, get_tokens
+from Brain.src.rising_plugin.csv_embed import csv_embed
+from Brain.src.rising_plugin.image_embedding import embed_image_text, query_image_text
+
+from Brain.src.logs import logger
+from Brain.src.model.basic_model import BasicModel
+from Brain.src.model.feedback_model import FeedbackModel
+from Brain.src.service.command_service import CommandService
+from Brain.src.service.contact_service import ContactsService
+from Brain.src.service.feedback_service import FeedbackService
+from Brain.src.service.llm.chat_service import ChatService
+from Brain.src.service.twilio_service import TwilioService
 
 from fastapi import APIRouter, Request, Depends
 

Brain/src/router/browser_router.py

@@ -1,10 +1,10 @@
 from fastapi import APIRouter, Request, Depends
 
-from src.common.assembler import Assembler
-from src.common.program_type import ProgramType
-from src.common.utils import parseUrlFromStr
-from src.model.requests.request_model import BrowserItem
-from src.service.browser_service import BrowserService
+from Brain.src.common.assembler import Assembler
+from Brain.src.common.program_type import ProgramType
+from Brain.src.common.utils import parseUrlFromStr
+from Brain.src.model.requests.request_model import BrowserItem
+from Brain.src.service.browser_service import BrowserService
 
 router = APIRouter()
 

Brain/src/service/browser_service.py

@@ -1,6 +1,6 @@
 """service to handle & process the browser"""
-from src.model.requests.request_model import BrowserItem
-from src.rising_plugin.risingplugin import getCompletionOnly
+from Brain.src.model.requests.request_model import BrowserItem
+from Brain.src.rising_plugin.risingplugin import getCompletionOnly
 
 
 class BrowserService:

Brain/src/service/command_service.py

@@ -1,7 +1,7 @@
 """service to manage command"""
 from typing import Any
 
-from src.commands.command import CommandRegistry
+from Brain.src.commands.command import CommandRegistry
 
 
 class CommandService:

Brain/src/service/contact_service.py

@@ -1,8 +1,8 @@
 """service to manage contacts"""
 from typing import List, Any
 
-from src.rising_plugin.csv_embed import get_embed
-from src.rising_plugin.pinecone_engine import (
+from Brain.src.rising_plugin.csv_embed import get_embed
+from Brain.src.rising_plugin.pinecone_engine import (
     get_pinecone_index_namespace,
     update_pinecone,
     init_pinecone,
@@ -11,8 +11,8 @@ from src.rising_plugin.pinecone_engine import (
     delete_all_pinecone,
 )
 
-from src.common.utils import PINECONE_INDEX_NAME
-from src.model.contact_model import ContactModel, ContactStatus
+from Brain.src.common.utils import PINECONE_INDEX_NAME
+from Brain.src.model.contact_model import ContactModel, ContactStatus
 
 
 class ContactsService:

Brain/src/service/feedback_service.py

@@ -3,7 +3,7 @@ from os import remove
 from firebase_admin import firestore
 
 
-from src.model.feedback_model import FeedbackModel
+from Brain.src.model.feedback_model import FeedbackModel
 
 """service to manage feedback collection"""
 

Brain/src/service/llm/chat_service.py

@@ -3,11 +3,11 @@ import time
 
 from openai.error import RateLimitError
 
-from src.common.utils import AGENT_NAME, DEFAULT_GPT_MODEL
-from src.rising_plugin.risingplugin import handle_chat_completion
-from src.logs import logger
-from src.model.chat_response_model import ChatResponseModel
-from src.model.message_model import MessageModel
+from Brain.src.common.utils import AGENT_NAME, DEFAULT_GPT_MODEL
+from Brain.src.rising_plugin.risingplugin import handle_chat_completion
+from Brain.src.logs import logger
+from Brain.src.model.chat_response_model import ChatResponseModel
+from Brain.src.model.message_model import MessageModel
 
 
 class ChatService:

Brain/src/service/twilio_service.py

@@ -4,9 +4,9 @@ from typing import Any
 
 from twilio.rest import Client
 
-from src.common.utils import ACCOUNT_SID, AUTH_TOKEN
-from src.logs import logger
-from src.model.sms_model import SMSModel
+from Brain.src.common.utils import ACCOUNT_SID, AUTH_TOKEN
+from Brain.src.logs import logger
+from Brain.src.model.sms_model import SMSModel
 
 
 class TwilioService:

Brain/tests/conftest.py

@@ -2,7 +2,7 @@ import pytest
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
-from src.firebase.firebase import initialize_app
+from Brain.src.firebase.firebase import initialize_app
 
 initialize_app()
 

Brain/tests/functional/test_recipes.py

@@ -1,10 +1,10 @@
 import re
 
-from src.firebase.firebase import initialize_app
-from src.rising_plugin.risingplugin import getCompletion, getTextFromImage
-from src.firebase.cloudmessage import send_message
-from src.firebase.cloudmessage import get_tokens
-from src.rising_plugin.image_embedding import query_image_text
+from Brain.src.firebase.firebase import initialize_app
+from Brain.src.rising_plugin.risingplugin import getCompletion, getTextFromImage
+from Brain.src.firebase.cloudmessage import send_message
+from Brain.src.firebase.cloudmessage import get_tokens
+from Brain.src.rising_plugin.image_embedding import query_image_text
 
 TEST_IAMGE_NAME = "0ddffe51-3763-48d9-ab74-2086de529217"
 TEST_UUID = "TEST_UUID"