update README

.env.template

@@ -0,0 +1,10 @@
+OPENAI_API_KEY=
+
+# Pinecone
+PINECONE_API_KEY=
+PINECONE_ENV=
+
+# search
+GOOGLE_API_KEY=
+GOOGLE_CX_ID=
+WOLFRAM_APP_ID=

README.md

@@ -60,7 +60,7 @@ direction and halt the process if it goes off the rails.**
 - PostgreSQL 14.4
 - Graphql
 - React 18
-- Framework to support remote and local AI models including GPT-4
+- Integrated with OpenAI GPT models
 - Plugin architecture to support extending agent functionality (e.g. web browsing, debugging code, etc)
 - Generic framework for vector database based agent memory
     - Pinecone
@@ -84,6 +84,26 @@ git clone https://github.com/kreneskyp/ix.git
 cd ix
 ```
 
+Setup config in `.env`
+
+```bash
+cp .env.template .env
+```
+
+```
+OPENAI_API_KEY=YOUR_KEY_HERE
+
+# Pinecone
+PINECONE_API_KEY=
+PINECONE_ENV=
+
+# search
+GOOGLE_API_KEY=
+GOOGLE_CX_ID=
+WOLFRAM_APP_ID=
+```
+
+
 Build and run the dev image:
 
 ```

ix/commands/tests/test_wolfram.py

@@ -0,0 +1,22 @@
+import pytest
+
+from ix.commands.wolfram import search_wolfram
+
+
+class TestSearchWolfram:
+    def test_search_wolfram_with_valid_query(self):
+        query = "What is the capital of France?"
+        results = search_wolfram(query)
+        assert len(results) > 0
+        assert all(isinstance(result, tuple) for result in results)
+
+    def test_search_wolfram_with_empty_query(self):
+        query = ""
+        with pytest.raises(ValueError):
+            search_wolfram(query)
+
+    def test_search_wolfram_with_missing_app_id(self, monkeypatch):
+        monkeypatch.delenv("WOLFRAM_APP_ID", raising=False)
+        query = "What is the capital of France?"
+        with pytest.raises(ValueError):
+            search_wolfram(query)

ix/commands/wolfram.py

@@ -0,0 +1,23 @@
+from typing import Callable, Any, List, Tuple
+import os
+import wolframalpha
+
+from ix.commands import command
+
+
+@command(name="search_wolfram", description="Search Wolfram.")
+def search_wolfram(search_string: str) -> List[Tuple[str, str]]:
+    """Searches Wolfram for the given search string and returns a list of (title, plaintext) tuples."""
+    app_id = os.environ.get("WOLFRAM_APP_ID")
+    if not app_id:
+        raise ValueError("WOLFRAM_APP_ID environment variable not set.")
+    if not search_string:
+        raise ValueError("search_string is required")
+    client = wolframalpha.Client(app_id)
+    print(search_string)
+    res = client.query(search_string)
+    results = []
+    for pod in res.pods:
+        for subpod in pod.subpods:
+            results.append((pod.title, subpod.plaintext))
+    return results

requirements.txt

@@ -19,3 +19,4 @@ ipython==8.12.0
 django_extensions==3.2.1
 pinecone-client==2.2.1
 tiktoken==0.3.3
+wolframalpha==5.0.0