update

.gitignore

@@ -5,4 +5,5 @@ __pycache__
 *.pyo
 
 .gradio
-local_path/
+local_path/
+images/

app.py

@@ -28,7 +28,7 @@ from kimi_dev.serve.templates import post_process,get_loc_prompt, clone_github_r
 
 TITLE = """<h1 align="left" style="min-width:200px; margin-top:0;">Chat with Kimi-Dev-72B🔥 </h1>"""
 DESCRIPTION_TOP = """<a href="https://github.com/MoonshotAI/Kimi-Dev" target="_blank">Kimi-Dev-72B</a> is a strong and open-source coding LLM for software engineering tasks."""
-USAGE_TOP = """Usage: 1. Input a Github url like "https://github.com/astropy/astropy" and submit it. \n2. Input your issue description and chat with Kimi-Dev-72B!"""
+USAGE_TOP = """Usage: 1. Input a Github url like "https://github.com/astropy/astropy" and a commit id and submit them. \n2. Input your issue description and chat with Kimi-Dev-72B!"""
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
 DEPLOY_MODELS = dict()
 logger = configure_logger()
@@ -88,6 +88,7 @@ def highlight_thinking(msg: str) -> str:
 def predict(
     text,
     url,
+    commit_hash,
     chatbot,
     history,
     top_p,
@@ -122,16 +123,20 @@ def predict(
 
     prompt = text
     repo_name = url.split("/")[-1]
+    print(url)
+    print(commit_hash)
 
     repo_path = './local_path/'+repo_name  # Local clone path
 
-    clone_github_repo(url, repo_path)
+    clone_github_repo(url, repo_path, commit_hash)
+    print("repo cloned")
     structure = build_repo_structure(repo_path)
+    # print("type(structure)",type(structure))
     string_struture = show_project_structure(structure)
+    # print("string_struturem,",string_struture)
 
     loc_prompt = get_loc_prompt(prompt,string_struture)
 
-
     messages = [
         {"role": "system", "content": "You are a helpful assistant."},
         {"role": "user", "content": loc_prompt}
@@ -247,6 +252,7 @@ def predict(
 def retry(
     text,
     url,
+    commit_hash,
     chatbot,
     history,
     top_p,
@@ -270,6 +276,7 @@ def retry(
     yield from predict(
         text,
         url,
+        commit_hash,
         chatbot,
         history,
         top_p,
@@ -284,6 +291,7 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
         history = gr.State([])
         input_text = gr.State()
         upload_url = gr.State()
+        commit_hash = gr.State()
 
         with gr.Row():
             gr.HTML(TITLE)
@@ -313,11 +321,13 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
                     retry_btn = gr.Button("🔄 Regenerate")
                     # del_last_btn = gr.Button("🗑️ Remove Last Turn")
             def respond(message):
-                return f"Url submitted!"
+                return f"Url and commit hash submitted!"
             with gr.Column():
                 url_box = gr.Textbox(label="Please input a Github url here",placeholder="Input your url", lines=1)
+                commit_hash_box = gr.Textbox(label="Please input a commit hash here",placeholder="Input your commit hash", lines=1)
+                # url_submit_btn = gr.Button("Submit")
                 url_submit_btn = gr.Button("Submit")
-                output = gr.Textbox(label="Submitted url")
+                output = gr.Textbox(label="Submitted url and commit")
                 url_submit_btn.click(fn=respond, inputs=upload_url, outputs=output)
 
                 # Parameter Setting Tab for control the generation parameters
@@ -332,13 +342,14 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
 
         gr.Examples(
             examples=get_examples(ROOT_DIR),
-            inputs=[url_box, text_box],
+            inputs=[url_box, text_box, commit_hash_box],
         )
         # gr.Markdown()
 
         input_widgets = [
             input_text,
             upload_url,
+            commit_hash,
             chatbot,
             history,
             top_p,
@@ -349,8 +360,8 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
 
         transfer_input_args = dict(
             fn=transfer_input,
-            inputs=[text_box, url_box],
-            outputs=[input_text, upload_url, text_box, upload_url, submit_btn],
+            inputs=[text_box, url_box,commit_hash_box],
+            outputs=[input_text, upload_url, text_box, commit_hash, submit_btn],
             show_progress=True,
         )
 
@@ -376,17 +387,17 @@ def main(args: argparse.Namespace):
     reload_javascript()
 
     favicon_path = os.path.join("kimi_dev/serve/assets/favicon.ico")
-    demo.queue().launch(
-        favicon_path=favicon_path,
-        server_name=args.ip,
-        server_port=args.port,
-    )
     # demo.queue().launch(
     #     favicon_path=favicon_path,
     #     server_name=args.ip,
     #     server_port=args.port,
-    #     share=True 
     # )
+    demo.queue().launch(
+        favicon_path=favicon_path,
+        server_name=args.ip,
+        server_port=args.port,
+        share=True 
+    )
 
 if __name__ == "__main__":
     args = parse_args()

kimi_dev/serve/examples.py

@@ -6,21 +6,24 @@ EXAMPLES_LIST = [
     [
         "https://github.com/astropy/astropy",
         "units.quantity_input decorator fails for constructors with type hinted return value -> None\n### Summary\r\nI am using the `units.quantity_input` decorator with typing hints for constructors, however when I add the correct return value for the constructor (`None`) then I get an exception, because `None` has no attribute `to`.\r\n\r\n### Reproducer\r\nThe issue can be reproduced with the following file:\r\n``` Python\r\nimport astropy.units as u\r\n\r\n\r\nclass PoC(object):\r\n\r\n    @u.quantity_input\r\n    def __init__(self, voltage: u.V) -> None:\r\n        pass\r\n\r\n\r\nif __name__ == '__main__':\r\n    poc = PoC(1.*u.V)\r\n```\r\nwhich results in the following error:\r\n```\r\n$ python3 poc.py\r\nTraceback (most recent call last):\r\n  File \"poc.py\", line 12, in <module>\r\n    poc = PoC(1.*u.V)\r\n  File \"/usr/lib64/python3.6/site-packages/astropy/utils/decorators.py\", line 868, in __init__\r\n    func = make_function_with_signature(func, name=name, **wrapped_args)\r\n  File \"/usr/lib64/python3.6/site-packages/astropy/units/decorators.py\", line 225, in wrapper\r\n    return return_.to(wrapped_signature.return_annotation)\r\nAttributeError: 'NoneType' object has no attribute 'to'\r\n```\r\n\r\nThis has been tested on Fedora 27 with python 3.6.3, astropy 2.0.2 and numpy 1.13.3 all from Fedora's repository.\r\n\r\n### Workaround\r\nThe issue can be circumvented by not adding the return type typing hint. Unfortunately, then a static type checker cannot infer that this function returns nothing.\r\n\r\n### Possible fix\r\nMaybe the decorator could explicitly check whether None is returned and then omit the unit check.\n\n\n",
+        "732d89c2940156bdc0e200bb36dc38b5e424bcba",
     ],
     [
         "https://github.com/sympy/sympy",
         "evalf does not call _imp_ recursively\nExample from https://stackoverflow.com/questions/41818842/why-cant-i-evaluate-a-composition-of-implemented-functions-in-sympy-at-a-point:\r\n\r\n```\r\n>>> from sympy.utilities.lambdify import implemented_function\r\n>>> f = implemented_function('f', lambda x: x ** 2)\r\n>>> g = implemented_function('g', lambda x: 2 * x)\r\n>>> print(f(  2 ).evalf())\r\n4.00000000000000\r\n>>> print(  g(2) .evalf())\r\n4.00000000000000\r\n>>> print(f(g(2)).evalf())\r\nf(g(2))\r\n```\r\n\r\nThe code for this is in `Function._eval_evalf`. It isn't calling evalf recursively on the return of `_imp_`. \n\n\n",
+        "d7c3045115693e887bcd03599b7ca4650ac5f2cb",
     ],
     [
         "https://github.com/matplotlib/matplotlib",
-        "[ENH]: ContourSet.set_paths\n### Problem\n\nTo get contour labelling working with its special transforms, Cartopy has a [workaround](https://github.com/SciTools/cartopy/blob/2ed668c17b4e52421f15c5be3761719c75c5311a/lib/cartopy/mpl/contour.py#L89-L108) where it replaces all the paths on the `ContourSet` with transformed versions.  This currently looks like\r\n\r\n```python\r\npaths = cs.get_paths()\r\npaths[:] = transformed_paths\r\n``` \r\n\r\nwhich doesn’t smell very good.\n\n### Proposed solution\n\nThe above would smell better as \r\n\r\n```python\r\ncs.set_paths(transformed_paths)\r\n``` \n\n\n"
+        "[ENH]: ContourSet.set_paths\n### Problem\n\nTo get contour labelling working with its special transforms, Cartopy has a [workaround](https://github.com/SciTools/cartopy/blob/2ed668c17b4e52421f15c5be3761719c75c5311a/lib/cartopy/mpl/contour.py#L89-L108) where it replaces all the paths on the `ContourSet` with transformed versions.  This currently looks like\r\n\r\n```python\r\npaths = cs.get_paths()\r\npaths[:] = transformed_paths\r\n``` \r\n\r\nwhich doesn’t smell very good.\n\n### Proposed solution\n\nThe above would smell better as \r\n\r\n```python\r\ncs.set_paths(transformed_paths)\r\n``` \n\n\n",
+        "2aee6ccd7c7e1f8d282c1e7579f4ee546b838542",
     ]
 ]
 
 
 def get_examples(root_dir: str = None):
     examples = []
-    for github_url, instance_id in EXAMPLES_LIST:
-        examples.append([github_url, instance_id])
+    for github_url, issue_description, commit_hash in EXAMPLES_LIST:
+        examples.append([github_url, issue_description, commit_hash])
 
     return examples

kimi_dev/serve/gradio_utils.py

@@ -12,11 +12,11 @@ import gradio as gr
 IMAGE_TOKEN = "<image>"
 
 
-def transfer_input(input_text, input_images):
+def transfer_input(input_text, input_url, input_commit_hash):
     """
     Transfer the input text and images to the input text and images.
     """
-    return (input_text, input_images, gr.update(value=""), gr.update(value=None), gr.Button(visible=True))
+    return (input_text, input_url, gr.update(value=""), input_commit_hash, gr.Button(visible=True))
 
 
 def delete_last_conversation(chatbot, history):

kimi_dev/serve/templates.py

@@ -26,14 +26,26 @@ def show_project_structure(structure, spacing=0) -> str:
 
     return pp_string
 
-def clone_github_repo(github_url, local_path):
-    """Clone GitHub repository to local path"""
+# def clone_github_repo(github_url, local_path):
+#     """Clone GitHub repository to local path"""
+#     try:
+#         subprocess.run(['git', 'clone', github_url, local_path], check=True)
+#         print(f"Successfully cloned repository to: {local_path}")
+#     except subprocess.CalledProcessError as e:
+#         print(f"Warning: Repository cloning may have failed: {e}")
+    
+def clone_github_repo(github_url, local_path, commit_hash=None):
+    """Clone GitHub repository to local path and optionally checkout specific commit"""
     try:
         subprocess.run(['git', 'clone', github_url, local_path], check=True)
         print(f"Successfully cloned repository to: {local_path}")
+        # If commit hash is provided, checkout to that specific commit
+        if commit_hash:
+            subprocess.run(['git', 'checkout', commit_hash], cwd=local_path, check=True)
+            print(f"Successfully checked out to commit: {commit_hash}")
     except subprocess.CalledProcessError as e:
-        print(f"Warning: Repository cloning may have failed: {e}")
-    
+        print(f"Warning: Repository cloning or checkout may have failed: {e}")
+
 def parse_python_file(file_path, file_content=None):
     """Parse a Python file to extract class and function definitions with their line numbers.
     :param file_path: Path to the Python file.