ruff y u so rough

Author: dot-agi

examples/ag2/async_human_input.ipynb

@@ -70,7 +70,9 @@
    "outputs": [],
    "source": [
     "agentops.init(auto_start_session=False)\n",
-    "tracer = agentops.start_trace(trace_name=\"AG2 Agent chat with Async Human Inputs\", tags=[\"ag2-chat-async-human-inputs\", \"agentops-example\"])"
+    "tracer = agentops.start_trace(\n",
+    "    trace_name=\"AG2 Agent chat with Async Human Inputs\", tags=[\"ag2-chat-async-human-inputs\", \"agentops-example\"]\n",
+    ")"
    ]
   },
   {

examples/ag2/tools_wikipedia_search.ipynb

@@ -71,7 +71,9 @@
    "outputs": [],
    "source": [
     "agentops.init(auto_start_session=False)\n",
-    "tracer = agentops.start_trace(trace_name=\"AG2 Wikipedia Search Tools\", tags=[\"ag2-wikipedia-search-tools\", \"agentops-example\"])"
+    "tracer = agentops.start_trace(\n",
+    "    trace_name=\"AG2 Wikipedia Search Tools\", tags=[\"ag2-wikipedia-search-tools\", \"agentops-example\"]\n",
+    ")"
    ]
   },
   {

examples/autogen/AgentChat.ipynb

@@ -99,7 +99,9 @@
    "source": [
     "# When initializing AgentOps, you can pass in optional tags to help filter sessions\n",
     "agentops.init(auto_start_session=False)\n",
-    "tracer = agentops.start_trace(trace_name=\"Microsoft Agent Chat Example\", tags=[\"autogen-chat\", \"microsoft-autogen\", \"agentops-example\"])"
+    "tracer = agentops.start_trace(\n",
+    "    trace_name=\"Microsoft Agent Chat Example\", tags=[\"autogen-chat\", \"microsoft-autogen\", \"agentops-example\"]\n",
+    ")"
    ]
   },
   {
@@ -133,7 +135,7 @@
    "outputs": [],
    "source": [
     "# Define model and API key\n",
-    "model_name = \"gpt-4-turbo\" # Or \"gpt-4o\" / \"gpt-4o-mini\" as per migration guide examples\n",
+    "model_name = \"gpt-4-turbo\"  # Or \"gpt-4o\" / \"gpt-4o-mini\" as per migration guide examples\n",
     "api_key = os.getenv(\"OPENAI_API_KEY\")\n",
     "\n",
     "# Create the model client\n",
@@ -142,26 +144,27 @@
     "# Create the agent that uses the LLM.\n",
     "assistant = AssistantAgent(\n",
     "    name=\"assistant\",\n",
-    "    system_message=\"You are a helpful assistant.\", # Added system message for clarity\n",
-    "    model_client=model_client\n",
+    "    system_message=\"You are a helpful assistant.\",  # Added system message for clarity\n",
+    "    model_client=model_client,\n",
     ")\n",
     "\n",
     "user_proxy_initiator = UserProxyAgent(\"user_initiator\")\n",
     "\n",
+    "\n",
     "async def main():\n",
-    "    termination = MaxMessageTermination(max_messages=2) \n",
+    "    termination = MaxMessageTermination(max_messages=2)\n",
     "\n",
     "    group_chat = RoundRobinGroupChat(\n",
-    "        [user_proxy_initiator, assistant], # Corrected: agents as positional argument\n",
-    "        termination_condition=termination\n",
+    "        [user_proxy_initiator, assistant],  # Corrected: agents as positional argument\n",
+    "        termination_condition=termination,\n",
     "    )\n",
-    "    \n",
+    "\n",
     "    chat_task = \"How can I help you today?\"\n",
     "    print(f\"User Initiator: {chat_task}\")\n",
     "\n",
     "    try:\n",
     "        stream = group_chat.run_stream(task=chat_task)\n",
-    "        await Console().run(stream) \n",
+    "        await Console().run(stream)\n",
     "        agentops.end_trace(tracer, end_state=\"Success\")\n",
     "\n",
     "    except StdinNotImplementedError:\n",
@@ -174,14 +177,16 @@
     "    finally:\n",
     "        await model_client.close()\n",
     "\n",
-    "if __name__ == '__main__':\n",
+    "\n",
+    "if __name__ == \"__main__\":\n",
     "    try:\n",
     "        loop = asyncio.get_running_loop()\n",
     "    except RuntimeError:\n",
     "        loop = None\n",
     "\n",
     "    if loop and loop.is_running():\n",
     "        import nest_asyncio\n",
+    "\n",
     "        nest_asyncio.apply()\n",
     "        asyncio.run(main())\n",
     "    else:\n",

examples/autogen/MathAgent.ipynb

@@ -96,7 +96,9 @@
    "outputs": [],
    "source": [
     "agentops.init(auto_start_session=False)\n",
-    "tracer = agentops.start_trace(trace_name=\"Microsoft Autogen Tool Example\", tags=[\"autogen-tool\", \"microsoft-autogen\", \"agentops-example\"])"
+    "tracer = agentops.start_trace(\n",
+    "    trace_name=\"Microsoft Autogen Tool Example\", tags=[\"autogen-tool\", \"microsoft-autogen\", \"agentops-example\"]\n",
+    ")"
    ]
   },
   {
@@ -139,6 +141,7 @@
     "\n",
     "Operator = Literal[\"+\", \"-\", \"*\", \"/\"]\n",
     "\n",
+    "\n",
     "def calculator(a: int, b: int, operator: Annotated[Operator, \"operator\"]) -> int:\n",
     "    if operator == \"+\":\n",
     "        return a + b\n",
@@ -151,29 +154,29 @@
     "    else:\n",
     "        raise ValueError(\"Invalid operator\")\n",
     "\n",
+    "\n",
     "async def main():\n",
     "    assistant = AssistantAgent(\n",
     "        name=\"Assistant\",\n",
     "        system_message=\"You are a helpful AI assistant. You can help with simple calculations. Return 'TERMINATE' when the task is done.\",\n",
     "        model_client=model_client,\n",
     "        tools=[calculator],\n",
-    "        reflect_on_tool_use=True\n",
+    "        reflect_on_tool_use=True,\n",
     "    )\n",
     "\n",
     "    initial_task_message = \"What is (1423 - 123) / 3 + (32 + 23) * 5?\"\n",
     "    print(f\"User Task: {initial_task_message}\")\n",
     "\n",
     "    try:\n",
     "        from autogen_core import CancellationToken\n",
-    "        \n",
+    "\n",
     "        response = await assistant.on_messages(\n",
-    "            [TextMessage(content=initial_task_message, source=\"user\")],\n",
-    "            CancellationToken()\n",
+    "            [TextMessage(content=initial_task_message, source=\"user\")], CancellationToken()\n",
     "        )\n",
-    "        \n",
+    "\n",
     "        final_response_message = response.chat_message\n",
     "        if final_response_message:\n",
-    "             print(f\"Assistant: {final_response_message.to_text()}\")\n",
+    "            print(f\"Assistant: {final_response_message.to_text()}\")\n",
     "        else:\n",
     "            print(\"Assistant did not provide a final message.\")\n",
     "\n",
@@ -188,14 +191,16 @@
     "    finally:\n",
     "        await model_client.close()\n",
     "\n",
-    "if __name__ == '__main__':\n",
+    "\n",
+    "if __name__ == \"__main__\":\n",
     "    try:\n",
     "        loop = asyncio.get_running_loop()\n",
     "    except RuntimeError:\n",
     "        loop = None\n",
     "\n",
     "    if loop and loop.is_running():\n",
     "        import nest_asyncio\n",
+    "\n",
     "        nest_asyncio.apply()\n",
     "        asyncio.run(main())\n",
     "    else:\n",

examples/crewai/job_posting.ipynb

@@ -32,16 +32,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "from crewai import Crew, Agent, Task\n",
-    "from crewai_tools.tools import WebsiteSearchTool, SerperDevTool, FileReadTool\n",
-    "import agentops\n",
-    "import os\n",
-    "from dotenv import load_dotenv"
-   ]
+   "source": "from crewai import Crew, Agent, Task\nfrom crewai_tools.tools import WebsiteSearchTool, SerperDevTool, FileReadTool\nimport agentops\nimport os\nfrom dotenv import load_dotenv\nfrom textwrap import dedent"
   },
   {
    "cell_type": "markdown",
@@ -123,83 +116,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "from textwrap import dedent\n",
-    "\n",
-    "\n",
-    "class Tasks:\n",
-    "    def research_company_culture_task(self, agent, company_description, company_domain):\n",
-    "        return Task(\n",
-    "            description=dedent(\n",
-    "                f\"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tAnalyze the provided company website and the hiring manager's company's domain {company_domain}, description: \"{company_description}\". Focus on understanding the company's culture, values, and mission. Identify unique selling points and specific projects or achievements highlighted on the site.\n",
-    "\t\t\t\t\t\t\t\tCompile a report summarizing these insights, specifically how they can be leveraged in a job posting to attract the right candidates.\"\"\"\n",
-    "            ),\n",
-    "            expected_output=dedent(\n",
-    "                \"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tA comprehensive report detailing the company's culture, values, and mission, along with specific selling points relevant to the job role. Suggestions on incorporating these insights into the job posting should be included.\"\"\"\n",
-    "            ),\n",
-    "            agent=agent,\n",
-    "        )\n",
-    "\n",
-    "    def research_role_requirements_task(self, agent, hiring_needs):\n",
-    "        return Task(\n",
-    "            description=dedent(\n",
-    "                f\"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tBased on the hiring manager's needs: \"{hiring_needs}\", identify the key skills, experiences, and qualities the ideal candidate should possess for the role. Consider the company's current projects, its competitive landscape, and industry trends. Prepare a list of recommended job requirements and qualifications that align with the company's needs and values.\"\"\"\n",
-    "            ),\n",
-    "            expected_output=dedent(\n",
-    "                \"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tA list of recommended skills, experiences, and qualities for the ideal candidate, aligned with the company's culture, ongoing projects, and the specific role's requirements.\"\"\"\n",
-    "            ),\n",
-    "            agent=agent,\n",
-    "        )\n",
-    "\n",
-    "    def draft_job_posting_task(self, agent, company_description, hiring_needs, specific_benefits):\n",
-    "        return Task(\n",
-    "            description=dedent(\n",
-    "                f\"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tDraft a job posting for the role described by the hiring manager: \"{hiring_needs}\". Use the insights on \"{company_description}\" to start with a compelling introduction, followed by a detailed role description, responsibilities, and required skills and qualifications. Ensure the tone aligns with the company's culture and incorporate any unique benefits or opportunities offered by the company.\n",
-    "\t\t\t\t\t\t\t\tSpecfic benefits: \"{specific_benefits}\"\"\"\n",
-    "            ),\n",
-    "            expected_output=dedent(\n",
-    "                \"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tA detailed, engaging job posting that includes an introduction, role description, responsibilities, requirements, and unique company benefits. The tone should resonate with the company's culture and values, aimed at attracting the right candidates.\"\"\"\n",
-    "            ),\n",
-    "            agent=agent,\n",
-    "        )\n",
-    "\n",
-    "    def review_and_edit_job_posting_task(self, agent, hiring_needs):\n",
-    "        return Task(\n",
-    "            description=dedent(\n",
-    "                f\"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tReview the draft job posting for the role: \"{hiring_needs}\". Check for clarity, engagement, grammatical accuracy, and alignment with the company's culture and values. Edit and refine the content, ensuring it speaks directly to the desired candidates and accurately reflects the role's unique benefits and opportunities. Provide feedback for any necessary revisions.\"\"\"\n",
-    "            ),\n",
-    "            expected_output=dedent(\n",
-    "                \"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tA polished, error-free job posting that is clear, engaging, and perfectly aligned with the company's culture and values. Feedback on potential improvements and final approval for publishing. Formated in markdown.\"\"\"\n",
-    "            ),\n",
-    "            agent=agent,\n",
-    "            output_file=\"job_posting.md\",\n",
-    "        )\n",
-    "\n",
-    "    def industry_analysis_task(self, agent, company_domain, company_description):\n",
-    "        return Task(\n",
-    "            description=dedent(\n",
-    "                f\"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tConduct an in-depth analysis of the industry related to the company's domain: \"{company_domain}\". Investigate current trends, challenges, and opportunities within the industry, utilizing market reports, recent developments, and expert opinions. Assess how these factors could impact the role being hired for and the overall attractiveness of the position to potential candidates.\n",
-    "\t\t\t\t\t\t\t\tConsider how the company's position within this industry and its response to these trends could be leveraged to attract top talent. Include in your report how the role contributes to addressing industry challenges or seizing opportunities.\"\"\"\n",
-    "            ),\n",
-    "            expected_output=dedent(\n",
-    "                \"\"\"\\\n",
-    "\t\t\t\t\t\t\t\tA detailed analysis report that identifies major industry trends, challenges, and opportunities relevant to the company's domain and the specific job role. This report should provide strategic insights on positioning the job role and the company as an attractive choice for potential candidates.\"\"\"\n",
-    "            ),\n",
-    "            agent=agent,\n",
-    "        )"
-   ]
+   "source": "class Tasks:\n    def research_company_culture_task(self, agent, company_description, company_domain):\n        return Task(\n            description=dedent(\n                f\"\"\"\\\n\t\t\t\t\t\t\t\tAnalyze the provided company website and the hiring manager's company's domain {company_domain}, description: \"{company_description}\". Focus on understanding the company's culture, values, and mission. Identify unique selling points and specific projects or achievements highlighted on the site.\n\t\t\t\t\t\t\t\tCompile a report summarizing these insights, specifically how they can be leveraged in a job posting to attract the right candidates.\"\"\"\n            ),\n            expected_output=dedent(\n                \"\"\"\\\n\t\t\t\t\t\t\t\tA comprehensive report detailing the company's culture, values, and mission, along with specific selling points relevant to the job role. Suggestions on incorporating these insights into the job posting should be included.\"\"\"\n            ),\n            agent=agent,\n        )\n\n    def research_role_requirements_task(self, agent, hiring_needs):\n        return Task(\n            description=dedent(\n                f\"\"\"\\\n\t\t\t\t\t\t\t\tBased on the hiring manager's needs: \"{hiring_needs}\", identify the key skills, experiences, and qualities the ideal candidate should possess for the role. Consider the company's current projects, its competitive landscape, and industry trends. Prepare a list of recommended job requirements and qualifications that align with the company's needs and values.\"\"\"\n            ),\n            expected_output=dedent(\n                \"\"\"\\\n\t\t\t\t\t\t\t\tA list of recommended skills, experiences, and qualities for the ideal candidate, aligned with the company's culture, ongoing projects, and the specific role's requirements.\"\"\"\n            ),\n            agent=agent,\n        )\n\n    def draft_job_posting_task(self, agent, company_description, hiring_needs, specific_benefits):\n        return Task(\n            description=dedent(\n                f\"\"\"\\\n\t\t\t\t\t\t\t\tDraft a job posting for the role described by the hiring manager: \"{hiring_needs}\". Use the insights on \"{company_description}\" to start with a compelling introduction, followed by a detailed role description, responsibilities, and required skills and qualifications. Ensure the tone aligns with the company's culture and incorporate any unique benefits or opportunities offered by the company.\n\t\t\t\t\t\t\t\tSpecfic benefits: \"{specific_benefits}\"\"\"\n            ),\n            expected_output=dedent(\n                \"\"\"\\\n\t\t\t\t\t\t\t\tA detailed, engaging job posting that includes an introduction, role description, responsibilities, requirements, and unique company benefits. The tone should resonate with the company's culture and values, aimed at attracting the right candidates.\"\"\"\n            ),\n            agent=agent,\n        )\n\n    def review_and_edit_job_posting_task(self, agent, hiring_needs):\n        return Task(\n            description=dedent(\n                f\"\"\"\\\n\t\t\t\t\t\t\t\tReview the draft job posting for the role: \"{hiring_needs}\". Check for clarity, engagement, grammatical accuracy, and alignment with the company's culture and values. Edit and refine the content, ensuring it speaks directly to the desired candidates and accurately reflects the role's unique benefits and opportunities. Provide feedback for any necessary revisions.\"\"\"\n            ),\n            expected_output=dedent(\n                \"\"\"\\\n\t\t\t\t\t\t\t\tA polished, error-free job posting that is clear, engaging, and perfectly aligned with the company's culture and values. Feedback on potential improvements and final approval for publishing. Formated in markdown.\"\"\"\n            ),\n            agent=agent,\n            output_file=\"job_posting.md\",\n        )\n\n    def industry_analysis_task(self, agent, company_domain, company_description):\n        return Task(\n            description=dedent(\n                f\"\"\"\\\n\t\t\t\t\t\t\t\tConduct an in-depth analysis of the industry related to the company's domain: \"{company_domain}\". Investigate current trends, challenges, and opportunities within the industry, utilizing market reports, recent developments, and expert opinions. Assess how these factors could impact the role being hired for and the overall attractiveness of the position to potential candidates.\n\t\t\t\t\t\t\t\tConsider how the company's position within this industry and its response to these trends could be leveraged to attract top talent. Include in your report how the role contributes to addressing industry challenges or seizing opportunities.\"\"\"\n            ),\n            expected_output=dedent(\n                \"\"\"\\\n\t\t\t\t\t\t\t\tA detailed analysis report that identifies major industry trends, challenges, and opportunities relevant to the company's domain and the specific job role. This report should provide strategic insights on positioning the job role and the company as an attractive choice for potential candidates.\"\"\"\n            ),\n            agent=agent,\n        )"
   },
   {
    "cell_type": "code",
@@ -276,4 +195,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}