chatGPT3.5.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "64c04dba",
   "metadata": {},
   "outputs": [],
   "source": [
    "import openai\n",
    "import json\n",
    "# openai.api_key =\"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\"\n",
    "openai.api_key =\"sk-H9kHXOCBqAYfeFLGA44666E7F80445EcAbD1F626F9BcD70e\"\n",
    "openai.api_base = \"https://gtapi.xiaoerchaoren.com:8932/v1\";\n",
    "# openai.api_key =\"sk-H9kHXOCBqAYfeFLGA44666E7F80445EcAbD1F626F9BcD70e\"\n",
    "completion = openai.ChatCompletion.create(\n",
    "    model=\"gpt-3.5-turbo\",\n",
    "    messages=[\n",
    "        {\"role\": \"system\", \"content\": \"作为一个社交媒体用户，用中文为以下帖子生成一条让人共情的回复(请用中文回复)\"},\n",
    "        {\"role\": \"user\", \"content\": \"试用期不买五险一金hr居然说.... hr说试用期不买五险一金，我说不买五险一金违法,hr说不买是违规不违法哦，没有上升到法律层面。yue了\"}\n",
    "    ]\n",
    ")\n",
    "print(completion)\n",
    "print(completion.choices[0].message['content'])\n",
    "# for i in completion:\n",
    "#   print(i)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "ea8e453b",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "5600\n",
      "5601\n",
      "5602\n",
      "5603\n",
      "5604\n",
      "5605\n",
      "5606\n",
      "5607\n",
      "5608\n",
      "5609\n",
      "5610\n",
      "5611\n",
      "5612\n",
      "5613\n",
      "5614\n",
      "5615\n",
      "5616\n",
      "5617\n",
      "5618\n",
      "5619\n",
      "5620\n",
      "5621\n",
      "5622\n",
      "5623\n",
      "5624\n",
      "5625\n",
      "5626\n",
      "5627\n",
      "5628\n",
      "5629\n",
      "5630\n",
      "5631\n",
      "5632\n",
      "5633\n",
      "5634\n",
      "5635\n",
      "5636\n",
      "5637\n",
      "5638\n",
      "5639\n",
      "5640\n",
      "5641\n",
      "5642\n",
      "5643\n",
      "5644\n",
      "5645\n",
      "5646\n",
      "5647\n",
      "5648\n",
      "5649\n",
      "5650\n",
      "5651\n",
      "5652\n",
      "5653\n",
      "5654\n",
      "5655\n",
      "5656\n",
      "5657\n",
      "5658\n",
      "5659\n",
      "5660\n",
      "5661\n",
      "5662\n",
      "5663\n",
      "5664\n",
      "5665\n",
      "5666\n",
      "5667\n",
      "5668\n",
      "5669\n",
      "5670\n",
      "5671\n",
      "5672\n",
      "5673\n",
      "5674\n",
      "5675\n",
      "5676\n",
      "5677\n",
      "5678\n",
      "5679\n",
      "5680\n",
      "5681\n",
      "5682\n",
      "5683\n",
      "5684\n",
      "5685\n",
      "5686\n",
      "5687\n",
      "5688\n",
      "5689\n",
      "5690\n",
      "5691\n",
      "5692\n",
      "5693\n",
      "5694\n",
      "5695\n",
      "5696\n",
      "5697\n",
      "5698\n",
      "5699\n",
      "100\n"
     ]
    }
   ],
   "source": [
    "import openai\n",
    "import json\n",
    "import pickle as pkl\n",
    "import time\n",
    "openai.api_key = \"sk-oUHVjesXusuAEckUA0A53eB3B57e4a07Be6297959547F275\"\n",
    "openai.api_base = \"https://gtapi.xiaoerchaoren.com:8932/v1\";\n",
    "responses = []\n",
    "import pandas as pd\n",
    "file = pd.read_csv(\"zctytlxz_postid_pre_final.csv\")\n",
    "for i in range(100):\n",
    "    j = i + 5600\n",
    "    completion = openai.ChatCompletion.create(\n",
    "        model=\"gpt-3.5-turbo\",\n",
    "        messages=[\n",
    "            {\"role\": \"system\", \"content\": \"作为一个社交媒体用户，在讨论职场问题的小组里看到以下帖子，用中文为它生成一条让人共情的回复\"},\n",
    "            {\"role\": \"user\", \"content\": file['postkey'][j][:6500]}\n",
    "        ]\n",
    "    )\n",
    "    st = completion.choices[0].message['content']\n",
    "    responses.append(st)\n",
    "    print(j)\n",
    "#     if i%100 == 0 and i != 0:\n",
    "#         with open(f\"result_zctytlxz_{j}_rt.pkl\".format(), \"wb\") as fa:\n",
    "#             pkl.dump(responses, fa)\n",
    "  # print(responses)\n",
    "with open(\"result_zctytlxz_5600.pkl\", \"wb\") as f:\n",
    "    pkl.dump(responses, f)\n",
    "# print(responses)\n",
    "print(len(responses))\n",
    "# print(completion.choices[0].message['content'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "17d5ee6d",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0\n"
     ]
    }
   ],
   "source": [
    "data = []\n",
    "with open(\"result_zctytlxz_8000.txt\",\"r\")as f:\n",
    "    for i in f.readlines():\n",
    "        data.append(i)\n",
    "print(len(data))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "3ac5a5b8",
   "metadata": {},
   "outputs": [],
   "source": [
    "i = 100000\n",
    "with open(f\"result_zctytlxz_{i}.pkl\".format(), \"wb\") as f:\n",
    "  pkl.dump(responses, f)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b4c51f20",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}