kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
ParthSareen 2025-09-23 17:31:56 -07:00
parent 9dbdcdf337
commit c5bbd4562a
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
examples/mcp-web-search-and-fetch.py
View File

@ -23,6 +23,7 @@ from ollama import Client
try:
# Preferred high-level API (if available)
from mcp.server.fastmcp import FastMCP # type: ignore
_FASTMCP_AVAILABLE = True
except Exception:
_FASTMCP_AVAILABLE = False
@ -33,7 +34,6 @@ if not _FASTMCP_AVAILABLE:
from mcp.server.stdio import stdio_server # type: ignore
client = Client()
@ -48,7 +48,7 @@ def _web_fetch_impl(url: str) -> Dict[str, Any]:
if _FASTMCP_AVAILABLE:
app = FastMCP("ollama-search-fetch")
app = FastMCP('ollama-search-fetch')
@app.tool()
def web_search(query: str, max_results: int = 3) -> Dict[str, Any]:
@ -79,11 +79,11 @@ if _FASTMCP_AVAILABLE:
return _web_fetch_impl(url=url)
if __name__ == "__main__":
if __name__ == '__main__':
app.run()
else:
server = Server("ollama-search-fetch") # type: ignore[name-defined]
server = Server('ollama-search-fetch') # type: ignore[name-defined]
@server.tool() # type: ignore[attr-defined]
async def web_search(query: str, max_results: int = 3) -> Dict[str, Any]:
@ -112,6 +112,5 @@ else:
async with stdio_server() as (read, write): # type: ignore[name-defined]
await server.run(read, write) # type: ignore[attr-defined]
if __name__ == "__main__":
if __name__ == '__main__':
asyncio.run(_main())