base_agent.py

from LLM import get_llm

Openai_Models = ["gpt-35-turbo-16k","gpt4-turbo-2024-04-29"]

class BaseAgent:
    # gpt-35-turbo-16k  gpt4-turbo-2024-04-29
    def __init__(self,model = "gpt-35-turbo-16k") -> None:
        self.model = model
        self.llm,self.img_generator = get_llm()
    
    def act(self, instruction):
        raise NotImplementedError
    

    def get_LLM_response(self,messages,**kwargs):
        if kwargs.get("model") == None:
            kwargs["model"] = self.model
        return self.llm.response(messages,**kwargs)
    
    def get_answer(self,messages,**kwargs):
        response = self.get_LLM_response(messages,model = self.model,**kwargs)
        return response
    
    async def get_LLM_response_async(self,messages,**kwargs):
        if kwargs.get("model") == None:
            kwargs["model"] = self.model
        response = await self.llm.response_async(messages,**kwargs)
        return response
    
    async def get_answer_async(self,messages,**kwargs):
        response = await self.get_LLM_response_async(messages,model = self.model,**kwargs)
        return response