main.py

import uuid
from langforge_os.utils import read_config, write_config

CONFIG_PATH = "config.yaml"


def get_model_details():
    config = read_config(CONFIG_PATH)
    model_name = config.get("model_name")
    base_url = config.get("base_url")
    api_key = config.get("api_key")
    if not model_name or not base_url or not api_key:
        model_name = input("Enter the model name: ")
        config["model_name"] = model_name
        base_url = input("Enter the base URL: ")
        config["base_url"] = base_url
        api_key = input("Enter the API key: ")
        config["api_key"] = api_key
        write_config(config, CONFIG_PATH)
    return model_name, base_url, api_key


model_name, base_url, api_key = get_model_details()

if __name__ == "__main__":

    from langforge_os.graph import app

    thread_id = str(uuid.uuid4())

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

    while True:
        user_input = input("You: ")

        for event in app.stream(
            {"messages": [("human", user_input)]}, stream_mode="values", config=config
        ):
            event["messages"][-1].pretty_print()