Working query.

- TODO: send query to agent.

Author: WhatTheFuzz

__init__.py

@@ -1,7 +1,7 @@
 from binaryninja import PluginCommand
 from . src.entry import check_function
 
-PluginCommand.register_for_address("What Does this Function Do?",
+PluginCommand.register_for_high_level_il_function("What Does this Function Do?",
                             "Checks OpenAI to see what this function does." \
                             "Requires an internet connection and an API key "
                             "saved under the environment variable "

src/agent.py

@@ -15,6 +15,11 @@
 
 class Agent:
 
+    question: str = '''
+    This is a function that was decompiled with Binary Ninja.
+    It is in Binary Ninja's IL_FORM. What does this function do?
+    '''
+
     # The maximum number of tokens that can be submitted to the engine.
     # See: https://beta.openai.com/docs/models/codex
     codex_limits: dict[str, int] = {
@@ -82,11 +87,9 @@ def generate_query(self, function: LowLevelILFunction |
         '''Generates a query string given a BNIL function. Reads the file
         prompt.txt and replaces the IL form with the name of the IL form.
         '''
-        prompt: str = ''
+        prompt: str = self.question
         # Read the prompt from the text file.
-        with open('prompt.txt', 'r') as prompt_file:
-            prompt = prompt_file.read()
-            prompt.replace('\{IL FORM\}', self.il_name[type(function)])
+        prompt = self.question.replace('IL_FORM', self.il_name[type(function)])
         # Add some new lines. Maybe not necessary.
         prompt += '\n\n'
         # Add the instructions to the prompt.

src/entry.py

@@ -3,3 +3,4 @@
 
 def check_function(bv: BinaryView, func: Function) -> bool:
     agent = Agent(function=func, engine='code-davinci-002')
+    print(agent.generate_query(func))

src/prompt.txt

@@ -1 +0,0 @@
-This is a function that was decompiled with Binary Ninja. It is in Binary Ninja's {IL FORM}. What does this function do?