matsuap commited on
Commit
66b4ff6
·
1 Parent(s): aa467c3

Add inner product and matrix multiplication tools using NumPy

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
- # server.py
2
  from fastmcp import FastMCP
 
3
 
4
  mcp = FastMCP("Demo 🚀")
5
 
@@ -12,6 +12,16 @@ def multiply(a: float, b: float) -> float:
12
  """Multiplies two numbers."""
13
  return a * b
14
 
 
 
 
 
 
 
 
 
 
 
15
  # Static resource
16
  @mcp.resource("config://version")
17
  def get_version():
 
 
1
  from fastmcp import FastMCP
2
+ import numpy as np
3
 
4
  mcp = FastMCP("Demo 🚀")
5
 
 
12
  """Multiplies two numbers."""
13
  return a * b
14
 
15
+ @mcp.tool()
16
+ def inner_product(a: list[float], b: list[float]) -> float:
17
+ """Calculates the inner product of two vectors."""
18
+ return np.dot(a, b)
19
+
20
+ @mcp.tool()
21
+ def matrix_multiply(a: list[list[float]], b: list[list[float]]) -> list[list[float]]:
22
+ """Multiplies two matrices."""
23
+ return np.matmul(a, b)
24
+
25
  # Static resource
26
  @mcp.resource("config://version")
27
  def get_version():