Merge pull request #52 from AnswerDotAI/cosette-contents-fix

cosette responses api fix

Author: ncoop57

nbs/00_core.ipynb

@@ -664,7 +664,7 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "conts = {'anthropic': cla.contents, 'openai': cos.contents}\n",
+    "conts = {'anthropic': cla.contents, 'openai': lambda r: getattr(r, 'output_text', r)}\n",
     "p = r'```(?:bash\\n|\\n)?([^`]+)```'\n",
     "def get_res(sage, q, provider, mode='default', verbosity=0):\n",
     "    if mode == 'command':\n",
@@ -1149,9 +1149,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "python3",
+   "display_name": "python",
    "language": "python",
-   "name": "python3"
+   "name": "python"
   }
  },
  "nbformat": 4,

shell_sage/core.py

@@ -249,7 +249,7 @@ def trace(msgs):
             if tool_use: print(f'Tool use: {tool_use.name}\nTool input: {tool_use.input}')
 
 # %% ../nbs/00_core.ipynb 33
-conts = {'anthropic': cla.contents, 'openai': cos.contents}
+conts = {'anthropic': cla.contents, 'openai': lambda r: getattr(r, 'output_text', r)}
 p = r'```(?:bash\n|\n)?([^`]+)```'
 def get_res(sage, q, provider, mode='default', verbosity=0):
     if mode == 'command':