Initial Kimi-K2-Instruct MCP server space

README.md

@@ -1,12 +1,12 @@
 ---
-title: Kimi K2 Mcp
-emoji: 🌍
-colorFrom: purple
-colorTo: yellow
+title: Kimi-K2 MCP
+emoji: 🌙
+colorFrom: indigo
+colorTo: pink
 sdk: gradio
 sdk_version: 6.5.1
 app_file: app.py
 pinned: false
+tags:
+  - mcp-server
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,186 @@
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+client = InferenceClient("moonshotai/Kimi-K2-Instruct")
+
+
+def generate_code(
+    prompt: str,
+    language: str = "python",
+    max_tokens: int = 2048,
+) -> str:
+    """
+    Generate code using Kimi-K2-Instruct (1T MoE model by Moonshot AI).
+
+    Args:
+        prompt: Description of the code you want to generate.
+            Be specific about what the code should do.
+        language: The programming language to generate code in
+            (e.g. python, javascript, rust, go, typescript).
+        max_tokens: Maximum number of tokens to generate (default 2048).
+
+    Returns:
+        The generated code as a string.
+    """
+    system_msg = (
+        f"You are an expert {language} programmer. "
+        "Return only clean, well-structured code with brief inline comments. "
+        "Do not include markdown fences or explanations outside the code."
+    )
+
+    response = client.chat_completion(
+        messages=[
+            {"role": "system", "content": system_msg},
+            {"role": "user", "content": prompt},
+        ],
+        max_tokens=max_tokens,
+    )
+    return response.choices[0].message.content
+
+
+def review_code(code: str) -> str:
+    """
+    Review code for bugs, security issues, and improvements using Kimi-K2-Instruct.
+
+    Args:
+        code: The source code to review.
+
+    Returns:
+        A concise review with issues found and suggested improvements.
+    """
+    response = client.chat_completion(
+        messages=[
+            {
+                "role": "system",
+                "content": (
+                    "You are a senior code reviewer. Analyze the code for: "
+                    "1) Bugs and logic errors "
+                    "2) Security vulnerabilities "
+                    "3) Performance issues "
+                    "4) Style and best practices. "
+                    "Be concise and actionable."
+                ),
+            },
+            {"role": "user", "content": f"Review this code:\n\n{code}"},
+        ],
+        max_tokens=2048,
+    )
+    return response.choices[0].message.content
+
+
+def explain_code(code: str) -> str:
+    """
+    Explain what a piece of code does in plain language using Kimi-K2-Instruct.
+
+    Args:
+        code: The source code to explain.
+
+    Returns:
+        A clear explanation of what the code does.
+    """
+    response = client.chat_completion(
+        messages=[
+            {
+                "role": "system",
+                "content": (
+                    "Explain the given code clearly and concisely. "
+                    "Cover what it does, how it works, and any notable patterns."
+                ),
+            },
+            {"role": "user", "content": f"Explain this code:\n\n{code}"},
+        ],
+        max_tokens=1024,
+    )
+    return response.choices[0].message.content
+
+
+def solve_problem(
+    problem: str,
+    max_tokens: int = 2048,
+) -> str:
+    """
+    Solve a coding or reasoning problem step by step using Kimi-K2-Instruct.
+
+    Args:
+        problem: The problem to solve. Can be algorithmic, mathematical,
+            or a general coding challenge.
+        max_tokens: Maximum number of tokens to generate (default 2048).
+
+    Returns:
+        A step-by-step solution with explanation and code.
+    """
+    response = client.chat_completion(
+        messages=[
+            {
+                "role": "system",
+                "content": (
+                    "You are an expert problem solver. "
+                    "Break down the problem, explain your approach, "
+                    "then provide a clean solution with code if applicable."
+                ),
+            },
+            {"role": "user", "content": problem},
+        ],
+        max_tokens=max_tokens,
+    )
+    return response.choices[0].message.content
+
+
+demo = gr.TabbedInterface(
+    [
+        gr.Interface(
+            fn=generate_code,
+            inputs=[
+                gr.Textbox(
+                    label="Prompt",
+                    placeholder="Describe the code you want...",
+                    lines=4,
+                ),
+                gr.Textbox(label="Language", value="python"),
+                gr.Slider(
+                    minimum=256, maximum=4096, value=2048, step=256,
+                    label="Max Tokens",
+                ),
+            ],
+            outputs=gr.Code(label="Generated Code"),
+            title="Code Generator",
+            api_name="generate_code",
+        ),
+        gr.Interface(
+            fn=review_code,
+            inputs=gr.Code(label="Code to Review", language="python", lines=15),
+            outputs=gr.Textbox(label="Review", lines=15),
+            title="Code Reviewer",
+            api_name="review_code",
+        ),
+        gr.Interface(
+            fn=explain_code,
+            inputs=gr.Code(label="Code to Explain", language="python", lines=15),
+            outputs=gr.Textbox(label="Explanation", lines=10),
+            title="Code Explainer",
+            api_name="explain_code",
+        ),
+        gr.Interface(
+            fn=solve_problem,
+            inputs=[
+                gr.Textbox(
+                    label="Problem",
+                    placeholder="Describe the problem to solve...",
+                    lines=6,
+                ),
+                gr.Slider(
+                    minimum=256, maximum=4096, value=2048, step=256,
+                    label="Max Tokens",
+                ),
+            ],
+            outputs=gr.Textbox(label="Solution", lines=15),
+            title="Problem Solver",
+            api_name="solve_problem",
+        ),
+    ],
+    ["Generate", "Review", "Explain", "Solve"],
+    title="Kimi-K2-Instruct Coding Assistant (MCP)",
+)
+
+if __name__ == "__main__":
+    demo.launch(mcp_server=True, ssr_mode=False)

requirements.txt

@@ -0,0 +1,2 @@
+gradio[mcp]>=5.0
+huggingface_hub>=0.25.0