Refactoring code

Author: hafeezali

examples/labeling/twitter_emotion_detection/tweet_emotion_detection_example.ipynb

@@ -0,0 +1,104 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "outputs": [],
+   "source": [
+    "from langroid.agent.chat_agent import ChatAgentConfig\n",
+    "from langroid.language_models.openai_gpt import OpenAIChatModel, OpenAIGPTConfig\n",
+    "from examples.labeling.twitter_emotion_detection.twitter_emotion_detection import TweetEmotionDetector"
+   ],
+   "metadata": {
+    "collapsed": false,
+    "ExecuteTime": {
+     "end_time": "2023-11-30T02:17:48.814122Z",
+     "start_time": "2023-11-30T02:17:48.804237Z"
+    }
+   },
+   "id": "46e7f6a620fa0233"
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "initial_id",
+   "metadata": {
+    "collapsed": true,
+    "ExecuteTime": {
+     "end_time": "2023-11-30T02:18:29.294787Z",
+     "start_time": "2023-11-30T02:18:29.292971Z"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "chat_agent_config = ChatAgentConfig(\n",
+    "    llm = OpenAIGPTConfig(\n",
+    "        chat_model = OpenAIChatModel.GPT4_TURBO,\n",
+    "    ),\n",
+    "    vecdb = None,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "outputs": [],
+   "source": [
+    "batch_base_prompt = \"\"\"\n",
+    "You are an expert at classifying tweets as neutral or one or more of the given emotions that best represent the mental state of the poster. Your job is to correctly label the provided input examples into one or more of the following categories: neutral, anger, anticipation, disgust, fear, joy, love, optimism, pessimism, sadness, surprise, trust.\n",
+    "\n",
+    "You will return the answer as a comma-separated list of labels sorted in alphabetical order for each of the input examples. For example: \n",
+    "\n",
+    "Sentiment[1]: label1, label2, label3\n",
+    "Sentiment[2]: label1, label2\n",
+    "Sentiment[3]: label1\n",
+    "\n",
+    "\"\"\""
+   ],
+   "metadata": {
+    "collapsed": false,
+    "ExecuteTime": {
+     "end_time": "2023-11-30T02:23:43.469457Z",
+     "start_time": "2023-11-30T02:23:43.457840Z"
+    }
+   },
+   "id": "46d1016f6bd65367"
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [
+    "tweet_emotion_detector = TweetEmotionDetector(\n",
+    "    chat_agent_config,\n",
+    "    \"\"\n",
+    ")"
+   ],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "ad4685a8dae4eabf"
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 2
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython2",
+   "version": "2.7.6"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}