add ruff

micvitc · Nov 14, 2024 · 4f27828 · 4f27828
1 parent 7b599b7
commit 4f27828
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 6 deletions.
diff --git a/langforge_os/graph.py b/langforge_os/graph.py
@@ -9,8 +9,6 @@
 from langforge_os.tools import tools
 from langchain_core.prompts import ChatPromptTemplate
 from langgraph.graph import START, END
-from pydantic import BaseModel
-from langchain_core.messages import ToolMessage, HumanMessage
 
 assistant_prompt = ChatPromptTemplate.from_messages(
     [

diff --git a/main.py b/main.py
@@ -1,16 +1,13 @@
 import uuid
 from langforge_os.graph import app
-from langchain_core.messages import ToolMessage
 
 
 if __name__ == "__main__":
-
     thread_id = str(uuid.uuid4())
 
     config = {"configurable": {"thread_id": thread_id}}
 
     while True:
-
         user_input = input("You: ")
 
         for event in app.stream(

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,6 +17,7 @@ tqdm = "^4.66.5"
 [tool.poetry.group.dev.dependencies]
 ipykernel = "^6.29.5"
 pytest = "^8.3.3"
+ruff = "^0.7.3"
 
 [build-system]
 requires = ["poetry-core"]