feat: code generator (wip)

eval/eval_widgets.txt

@@ -34,3 +34,4 @@ display_uniswap
 fetch_nfts_owned_by_address_or_domain
 fetch_nfts_owned_by_user
 fetch_link_suggestion
+generate_js_code

knowledge_base/widgets.yaml

@@ -1,22 +1,22 @@
-- _name_: display_transfer
-  description: Transfer a token from a user's wallet to another address
-  parameters:
-    properties:
-      address:
-        description: Transfer recipient address.
-        type: string
-      amount:
-        description: Quantity to transfer.
-        type: string
-      token:
-        description: Symbol of the token being transferred.
-        type: string
-    required:
-    - token
-    - amount
-    - address
-    type: object
-  return_value_description: ''
+# - _name_: display_transfer
+#   description: Transfer a token from a user's wallet to another address
+#   parameters:
+#     properties:
+#       address:
+#         description: Transfer recipient address.
+#         type: string
+#       amount:
+#         description: Quantity to transfer.
+#         type: string
+#       token:
+#         description: Symbol of the token being transferred.
+#         type: string
+#     required:
+#     - token
+#     - amount
+#     - address
+#     type: object
+#   return_value_description: ''
 - _name_: fetch_nft_buy_asset
   description: Buy an NFT asset of a collection on the OpenSea marketplace, given
     its network, address, and token ID. Don't use this if we don't have the collection
@@ -854,4 +854,15 @@
     - amount
     - vault
     type: object
-  return_value_description: ''
+  return_value_description: ''
+- _name_: generate_js_code
+  description: generate code for the user based on the query; this widget should always be chosen for anything relating to code within a query
+  parameters:
+    properties:
+      query:
+        description: a standalone question representing the user's intent to perform an action via code
+        type: string
+    required:
+    - query
+    type: object
+  return_value_description: formatted javascript code

tools/__init__.py

@@ -6,4 +6,5 @@
 from . import index_app_info
 from . import index_api_tool
 from . import app_usage_guide
-from . import index_link_suggestion
+from . import index_link_suggestion
+from . import generate_js_code

tools/generate_js_code.py

@@ -0,0 +1,71 @@
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from pydantic import Extra
+
+import registry
+import streaming
+from .base import BaseTool, BASE_TOOL_DESCRIPTION_TEMPLATE
+
+
+TEMPLATE = '''You are an expert Web3 developer well versed in using JS to interact with the ecosystem, you will help the user perform actions based on their request by generating functional JS code
+
+# INSTRUCTIONS
+- Assume user wallet already connected to browser so never ask for a private key, Infura project ID, or any credentials
+- Print out transaction hash if applicable
+- Always use ethers.js
+- Assume there is an ethers.js provider and signer available and can be provided to the function or code
+- The code should return a function or a promise that can be called to perform the action
+- Your final output should be a formatted JS code function with comments, which can be run on a frontend; don't include anything else for now (ie: messages that precede the code, etc.)
+
+---
+User: {question}
+Assistant:'''
+
+
+@registry.register_class
+class GenerateJSCodeTool(BaseTool):
+    """Tool for generating code to perform a user request."""
+
+    _chain: LLMChain
+
+    class Config:
+        """Configuration for this pydantic object."""
+        extra = Extra.allow
+
+    def __init__(
+            self,
+            *args,
+            **kwargs
+    ) -> None:
+        prompt = PromptTemplate(
+            input_variables=["question"],
+            template=TEMPLATE,
+        )
+        new_token_handler = kwargs.get('new_token_handler')
+        chain = streaming.get_streaming_chain(prompt, new_token_handler)
+        description = BASE_TOOL_DESCRIPTION_TEMPLATE.format(
+            tool_description="generate code based on the user query",
+            input_description="a standalone query where user wants to generate code to perform an action",
+            output_description="a JSON object with a code field, which contains a formatted JS code function with comments, which can be run on a frontend; don't include anything else for now (ie: messages that precede the code, etc.)"
+        )
+
+        super().__init__(
+            *args,
+            _chain=chain,
+            description=description,
+            **kwargs
+        )
+
+    def _run(self, query: str) -> str:
+        example = {
+            "question": query,
+            "stop": "User",
+        }
+        result = self._chain.run(example)
+        print('result in generate_js_code', result)
+
+        return result
+
+    async def _arun(self, query: str) -> str:
+        raise NotImplementedError(
+            f"{self.__class__.__name__} does not support async")