docs: add examples based on uv run scripts (#154)

Author: andriisoldatenko

examples/README.md

@@ -0,0 +1,22 @@
+## Collection of Python galileo examples
+
+### Preconditions
+
+Install `uv`, we use inline dependency inside scripts.
+
+### How to use/run?
+
+First of all create `.env` file and add required env vars based on `.env.sample`.
+
+Then just run `uv`:
+
+```bash
+uv run --env-file=examples/langgraph/.env examples/langgraph/with_openai.py
+```
+
+or
+
+#### [basic_langgraph.py]
+```bash
+uv run --env-file=examples/langgraph/.env examples/langgraph/basic_langgraph.py
+```

examples/langgraph/.env.sample

@@ -0,0 +1,4 @@
+GALILEO_API_KEY=
+GALILEO_PROJECT=
+GALILEO_LOG_STREAM=
+OPENAI_API_KEY=

examples/langgraph/basic_langgraph.py

@@ -0,0 +1,48 @@
+# /// script
+# requires-python = ">=3.10"
+# dependencies = [
+#     "galileo",
+#     "langgraph",
+#     "langsmith",
+#     "langchain",
+#     "grandalf", # for printing graph in ascii
+# ]
+# ///
+# from dotenv import load_dotenv; load_dotenv()
+
+from typing import Annotated
+
+from langchain_core.messages import AIMessage
+from langgraph.graph import END, START, StateGraph
+from langgraph.graph.message import add_messages
+from typing_extensions import TypedDict
+
+from galileo.handlers.langchain import GalileoCallback
+
+
+class State(TypedDict):
+    # Messages have the type "list". The `add_messages` function
+    # in the annotation defines how this state key should be updated
+    # (in this case, it appends messages to the list, rather than overwriting them)
+    messages: Annotated[list, add_messages]
+
+
+def node(state: State):
+    messages = state["messages"]
+    new_message = AIMessage("Hello!")
+
+    return {"messages": messages + [new_message], "extra_field": 10}
+
+
+def node2(state: State):
+    return {"messages": state["messages"]}
+
+
+graph_builder = StateGraph(State)
+graph_builder.add_node("node_name", node)
+graph_builder.add_edge(START, "node_name")
+graph_builder.add_edge("node_name", END)
+graph = graph_builder.compile()
+
+graph.get_graph().print_ascii()
+graph.invoke({"messages": [{"role": "user", "content": "hi!"}]}, config={"callbacks": [GalileoCallback()]})

examples/langgraph/with_openai.py

@@ -0,0 +1,45 @@
+# /// script
+# requires-python = ">=3.10"
+# dependencies = [
+#     "galileo",
+#     "langgraph",
+#     "langsmith",
+#     "langchain[openai]",
+#     "grandalf", # for printing graph in ascii
+# ]
+# ///
+from typing import Annotated
+
+from langchain_openai import ChatOpenAI
+from langgraph.graph import START, StateGraph
+from langgraph.graph.message import add_messages
+from typing_extensions import TypedDict
+
+from galileo.handlers.langchain import GalileoCallback
+
+
+class State(TypedDict):
+    # Messages have the type "list". The `add_messages` function
+    # in the annotation defines how this state key should be updated
+    # (in this case, it appends messages to the list, rather than overwriting them)
+    messages: Annotated[list, add_messages]
+
+
+llm = ChatOpenAI(model="gpt-4")
+
+
+def chatbot(state: State):
+    return {"messages": [llm.invoke(state["messages"])]}
+
+
+graph_builder = StateGraph(State)
+# The first argument is the unique node name
+# The second argument is the function or object that will be called whenever
+# the node is used.
+graph_builder.add_node("chatbot", chatbot)
+graph_builder.add_edge(START, "chatbot")
+graph = graph_builder.compile()
+
+graph.get_graph().print_ascii()
+
+graph.invoke({"messages": [{"role": "user", "content": "hi!"}]}, {"callbacks": [GalileoCallback()]})