Criando conversas

Trabalhando com Llama 3

Imtihan Ahmed

Machine Learning Engineer

Mantendo o contexto

Fluxo para manter contexto - etapa 1

Mantendo o contexto

Fluxo para manter contexto - etapa 2

Mantendo o contexto

Fluxo para manter contexto - etapa 3

Mantendo o contexto

Fluxo para manter contexto - etapa 4

Mantendo o contexto

Fluxo para manter contexto - etapa 5

Acompanhe o histórico do chat com a classe Conversation

Classe Conversation

Pode guardar o histórico de mensagens anteriores

class Conversation:

    def __init__(self, llm: Llama, system_prompt='', history=[]):
        self.llm = llm
        self.system_prompt = system_prompt

        self.history = [{"role": "system", "content": self.system_prompt}] + history

Classe Conversation

Pode guardar o histórico de mensagens anteriores

class Conversation:
    def __init__(self, llm: Llama, system_prompt='', history=[]):
        self.llm = llm
        self.system_prompt = system_prompt
        self.history = [{"role": "system", "content": self.system_prompt}] + history


    def create_completion(self, user_prompt=''):

        self.history.append({"role": "user", "content": user_prompt}) # Adiciona a entrada

        output = self.llm.create_chat_completion(messages=self.history)

        conversation_result = output['choices'][0]['message']
        self.history.append(conversation_result) # Adiciona a saída
        return conversation_result['content'] # Retorna a saída

Executando uma conversa multi-turnos

conversation = Conversation(llm, system_prompt="You are a virtual travel assistant 
                                                helping with planning trips.")


response1 = conversation.create_completion("What are some destinations in France for a 
                                            short weekend break?")

print(f"Response 1: {response1}")


response2 = conversation.create_completion("How about Spain?")


print(f"Response 2: {response2}")

Executando uma conversa multi-turnos

print(f"Response 1: {response1}")

print(f"Response 2: {response2}")

Saída

Executando uma conversa multi-turnos

print(f"Response 1: {response1}")

print(f"Response 2: {response2}")

Saída com partes destacadas

Vamos praticar!

Trabalhando com Llama 3