Skip to content

Commit f0b5340

Browse files
committed
Apply code formatting (isort, black)
1 parent 31ba74b commit f0b5340

File tree

1 file changed

+5
-13
lines changed
  • contributing/samples/mcp_postgres_agent

1 file changed

+5
-13
lines changed

contributing/samples/mcp_postgres_agent/agent.py

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,10 @@
1515
import os
1616

1717
from dotenv import load_dotenv
18-
from google.adk.models.gemma_llm import Gemma
19-
from google.genai.types import GenerateContentConfig
20-
2118
from google.adk.agents.llm_agent import LlmAgent
2219
from google.adk.tools.mcp_tool import StdioConnectionParams
2320
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
21+
from google.genai.types import GenerateContentConfig
2422
from mcp import StdioServerParameters
2523

2624
load_dotenv()
@@ -35,19 +33,13 @@
3533
"Use the provided tools to query, manage, and interact with "
3634
"the PostgreSQL database. Ask clarifying questions when unsure."
3735
),
38-
3936
tools=[
4037
MCPToolset(
4138
connection_params=StdioConnectionParams(
4239
server_params=StdioServerParameters(
43-
command='uvx',
44-
args=[
45-
'postgres-mcp',
46-
'--access-mode=unrestricted'
47-
],
48-
env={
49-
"DATABASE_URI": POSTGRES_CONNECTION_STRING
50-
}
40+
command="uvx",
41+
args=["postgres-mcp", "--access-mode=unrestricted"],
42+
env={"DATABASE_URI": POSTGRES_CONNECTION_STRING},
5143
),
5244
timeout=60,
5345
),
@@ -56,5 +48,5 @@
5648
generate_content_config=GenerateContentConfig(
5749
temperature=1.0,
5850
top_p=0.95,
59-
)
51+
),
6052
)

0 commit comments

Comments
 (0)