Làm việc với Llama 3
Imtihan Ahmed
Machine Learning Engineer





Conversationclass Conversation:def __init__(self, llm: Llama, system_prompt='', history=[]): self.llm = llm self.system_prompt = system_promptself.history = [{"role": "system", "content": self.system_prompt}] + history
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}] + historydef create_completion(self, user_prompt=''):self.history.append({"role": "user", "content": user_prompt}) # Thêm đầu vàooutput = self.llm.create_chat_completion(messages=self.history)conversation_result = output['choices'][0]['message'] self.history.append(conversation_result) # Thêm đầu ra return conversation_result['content'] # Trả về đầu ra
conversation = Conversation(llm, system_prompt="Bạn là một trợ lý du lịch ảo hỗ trợ lập kế hoạch chuyến đi.")response1 = conversation.create_completion("Có điểm đến nào ở Pháp cho kỳ nghỉ cuối tuần ngắn không?")print(f"Response 1: {response1}")response2 = conversation.create_completion("Còn Tây Ban Nha thì sao?")print(f"Response 2: {response2}")
print(f"Response 1: {response1}")
print(f"Response 2: {response2}")

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

Làm việc với Llama 3