本地大模型编程实战(07)自制聊天机器人(1)

def chat(model_name):
    model = ChatOllama(model=model_name,temperature=0.3,verbose=True)
    response = model.invoke([HumanMessage(content="Hi! I'm Bob")])
    print(f'chat_with_no_memory:\n{response.content}')

    # We can see that it doesn't take the previous conversation turn into context, and cannot answer the question. This makes for a terrible chatbot experience!
    response = model.invoke([HumanMessage(content="What's my name?")])
    print(f'chat_with_no_memory 2:\n{response.content}')

def chat_with_memory(model_name):
    '''具有记忆功能'''
    model = ChatOllama(model=model_name,temperature=0.3,verbose=True)
    response = model.invoke(
        [
            HumanMessage(content="Hi! I'm Bob"),
            AIMessage(content="Hello Bob! How can I assist you today?"),
            HumanMessage(content="What's my name?"),
        ]
    )
    print(f'chat_with_memory:\n{response.content}')

def build_app(model_name):

    model = ChatOllama(model=model_name,temperature=0.3,verbose=True)

    # Define the function that calls the model
    def call_model(state: MessagesState):
        response = model.invoke(state["messages"])
        return {"messages": response}

    # Define a new graph
    workflow = StateGraph(state_schema=MessagesState)

    # Define the (single) node in the graph
    workflow.add_edge(START, "model")
    workflow.add_node("model", call_model)

    # Add memory
    memory = MemorySaver()
    app = workflow.compile(checkpointer=memory)
    return app

def message_persistence():
    app = build_app()
    
    config = {"configurable": {"thread_id": "abc123"}}

    query = "Hi! I'm Bob."

    input_messages = [HumanMessage(query)]
    output = app.invoke({"messages": input_messages}, config)
    print(output["messages"][-1].pretty_print())  # output contains all messages in state

    query = "What's my name?"

    input_messages = [HumanMessage(query)]
    output = app.invoke({"messages": input_messages}, config)
    print(output["messages"][-1].pretty_print())

    # different thread_id
    config = {"configurable": {"thread_id": "abc234"}}

    input_messages = [HumanMessage(query)]
    output = app.invoke({"messages": input_messages}, config)
    print(output["messages"][-1].pretty_print())

    config = {"configurable": {"thread_id": "abc123"}}

    input_messages = [HumanMessage(query)]
    output = app.invoke({"messages": input_messages}, config)
    print(output["messages"][-1].pretty_print())

================================== Ai Message ==================================

Nice to meet you, Bob! How's your day going so far? Is there something on your mind that you'd like to chat about or is this just a friendly hello?
None
================================== Ai Message ==================================

Your name is Bob! We established that right at the beginning of our conversation.
None
================================== Ai Message ==================================

I'm a large language model, I don't have any information about your personal identity or name. ... , I'd be happy to chat with you!
None
================================== Ai Message ==================================

Your name is Bob, as we've already discussed!
None

================================== Ai Message ==================================

<think>

</think>

Hello, Bob! How can I assist you today? 😊
None
================================== Ai Message ==================================

<think>
Alright, the user greeted me as Bob and mentioned they're Bob. Now, they asked, "What's my name?" 

...

I should make sure the response is warm and approachable, letting them know I'm here to help.
</think>

Hello! My name is Bob. How can I assist you today? 😊
None
================================== Ai Message ==================================

<think>

</think>

Hi! I don’t have access to personal information like names or identities. Could you ask something else I can help you with?
None
================================== Ai Message ==================================

<think>
Okay, so the user has been interacting with me as Bob multiple times now. They asked "What's my name?" again after I told them it was Bob.

...

I should make sure the response is clear and friendly, letting them know I'm here to help without overcomplicating things.
</think>

It seems like you might be asking about my name, but as an AI, I don't have personal experiences or feelings. My "name" is a label you've given me in our conversation. How can I assist you further? 😊
None