fix lint

Signed-off-by: Jack Luar <[email protected]>

Author: luarss

backend/src/agents/retriever_mcp.py

@@ -5,14 +5,14 @@
 from langgraph.graph import END
 from .retriever_typing import AgentState
 from langchain.prompts import ChatPromptTemplate
-from ..prompts.prompt_templates import run_orfs_prompt_template
 from langchain_core.tools.base import ToolException
 from langchain_google_vertexai import ChatVertexAI
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_ollama import ChatOllama
 
 from ..prompts.tool_examples import mcp_system_prompt, few_shot_examples
 
+
 class MCP:
     llm: ChatVertexAI | ChatGoogleGenerativeAI | ChatOllama
     workflow: Any
@@ -24,16 +24,16 @@ def mcp_agent(self, state: AgentState) -> dict[str, list[Any]]:
         custom_tools = get_tools()
         model = self.llm.bind_tools(custom_tools)
 
-        orfs_tmp = ChatPromptTemplate.from_messages([
-            ("system", mcp_system_prompt),
-            *few_shot_examples,
-            (
-            "user",
-            "Previous conversation:\n{chat_history}\n\nCurrent question:\n{question}",
-            )
-        ])
-
-        #orfs_tmp = ChatPromptTemplate.from_template(run_orfs_prompt_template)
+        orfs_tmp = ChatPromptTemplate.from_messages(
+            [
+                ("system", mcp_system_prompt),
+                *few_shot_examples,
+                (
+                    "user",
+                    "Previous conversation:\n{chat_history}\n\nCurrent question:\n{question}",
+                ),
+            ]
+        )
 
         run_orfs_chain = orfs_tmp | model
 

backend/src/openroad_mcp/server/orfs/orfs_base.py

@@ -113,7 +113,7 @@ def metrics(cmd: str) -> str:
         ORFS.server._check_configuration()
         ORFS.server._command("metadata")
 
-        return f"finished metadata"
+        return "finished metadata"
 
     @staticmethod
     @ORFS.mcp.tool

backend/src/prompts/tool_examples.py

@@ -10,90 +10,71 @@
 
 few_shot_examples = [
     ### Start
-
     ### get_platforms
     {
         "role": "user",
-        "content": "Get list of available platforms and select default plaform."
+        "content": "Get list of available platforms and select default plaform.",
     },
     {
         "role": "assistant",
         "content": """{{
   "name": "get_platforms",
   "args": {{}}
-}}"""
+}}""",
     },
     ###
-
     ### get_designs
     {
         "role": "user",
-        "content": "Get list of available designs and select default design."
+        "content": "Get list of available designs and select default design.",
     },
     {
         "role": "assistant",
         "content": """{{
   "name": "get_designs",
   "args": {{}}
-}}"""
+}}""",
     },
     ###
-
     ### make
-    {
-        "role": "user",
-        "content": "make clean_all"
-    },
+    {"role": "user", "content": "make clean_all"},
     {
         "role": "assistant",
         "content": """{{
   "name": "make",
   "args": {{"cmd": "clean_all"}}
-}}"""
+}}""",
     },
     ###
-
     ### get_stage_names
-    {
-        "role": "user",
-        "content": "Get list of stage names available to run."
-    },
+    {"role": "user", "content": "Get list of stage names available to run."},
     {
         "role": "assistant",
         "content": """{{
   "name": "get_stage_names",
   "args": {{}}
-}}"""
+}}""",
     },
     ###
-
     ### jump
-    {
-        "role": "user",
-        "content": "jump floorplan"
-    },
+    {"role": "user", "content": "jump floorplan"},
     {
         "role": "assistant",
         "content": """{{
   "name": "jump",
   "args": {{"stage": "floorplan"}}
-}}"""
+}}""",
     },
     ###
-
     ### jump
-    {
-        "role": "user",
-        "content": "step"
-    },
+    {"role": "user", "content": "step"},
     {
         "role": "assistant",
         "content": """{{
   "name": "step",
   "args": {{}}
-}}"""
+}}""",
     },
     ###
-
     ### End
 ]