From 76aacaffdcde0fc76629c35881d33109966c8c93 Mon Sep 17 00:00:00 2001 From: Joel Alexander Date: Mon, 28 Aug 2023 22:44:58 -0700 Subject: [PATCH] codestyle --- parea/cookbook/tracing_with_Parea_sdk.ipynb | 126 ++++++++++---------- 1 file changed, 63 insertions(+), 63 deletions(-) diff --git a/parea/cookbook/tracing_with_Parea_sdk.ipynb b/parea/cookbook/tracing_with_Parea_sdk.ipynb index 914e2452..cb827998 100644 --- a/parea/cookbook/tracing_with_Parea_sdk.ipynb +++ b/parea/cookbook/tracing_with_Parea_sdk.ipynb @@ -113,50 +113,50 @@ "# We pass the deployment_id and the required inputs to the completion function along with the trace_id\n", "@trace\n", "def argument_generator(query: str, additional_description: str = \"\") -> str:\n", - " return p.completion(\n", - " Completion(\n", - " deployment_id=\"p-Ar-Oi14-nBxHUiradyql9\",\n", - " llm_inputs={\n", - " \"additional_description\": additional_description,\n", - " \"date\": f\"{datetime.now()}\",\n", - " \"query\": query,\n", - " },\n", - " )\n", - " ).content\n", + " return p.completion(\n", + " Completion(\n", + " deployment_id=\"p-Ar-Oi14-nBxHUiradyql9\",\n", + " llm_inputs={\n", + " \"additional_description\": additional_description,\n", + " \"date\": f\"{datetime.now()}\",\n", + " \"query\": query,\n", + " },\n", + " )\n", + " ).content\n", "\n", "\n", "@trace\n", "def critic(argument: str) -> str:\n", - " return p.completion(\n", - " Completion(\n", - " deployment_id=\"p-W2yPy93tAczYrxkipjli6\",\n", - " llm_inputs={\"argument\": argument},\n", - " )\n", - " ).content\n", + " return p.completion(\n", + " Completion(\n", + " deployment_id=\"p-W2yPy93tAczYrxkipjli6\",\n", + " llm_inputs={\"argument\": argument},\n", + " )\n", + " ).content\n", "\n", "\n", "@trace\n", "def refiner(query: str, additional_description: str, current_arg: str, criticism: str) -> str:\n", - " return p.completion(\n", - " Completion(\n", - " deployment_id=\"p-8Er1Xo0GDGF2xtpmMOpbn\",\n", - " llm_inputs={\n", - " \"additional_description\": additional_description,\n", - " \"date\": f\"{datetime.now()}\",\n", - " \"query\": query,\n", - " \"current_arg\": current_arg,\n", - " \"criticism\": criticism,\n", - " },\n", - " )\n", - " ).content\n", + " return p.completion(\n", + " Completion(\n", + " deployment_id=\"p-8Er1Xo0GDGF2xtpmMOpbn\",\n", + " llm_inputs={\n", + " \"additional_description\": additional_description,\n", + " \"date\": f\"{datetime.now()}\",\n", + " \"query\": query,\n", + " \"current_arg\": current_arg,\n", + " \"criticism\": criticism,\n", + " },\n", + " )\n", + " ).content\n", "\n", "\n", "# This is the parent function which orchestrates the chaining. We'll define our trace_id and trace_name here\n", "@trace\n", "def argument_chain(query: str, additional_description: str = \"\") -> str:\n", - " argument = argument_generator(query, additional_description)\n", - " criticism = critic(argument)\n", - " return refiner(query, additional_description, argument, criticism)" + " argument = argument_generator(query, additional_description)\n", + " criticism = critic(argument)\n", + " return refiner(query, additional_description, argument, criticism)" ] }, { @@ -184,8 +184,8 @@ "outputs": [], "source": [ "result = argument_chain(\n", - " \"Whether moonshine is good for you.\",\n", - " additional_description=\"Provide a concise, few sentence argument on why moonshine is good for you.\",\n", + " \"Whether moonshine is good for you.\",\n", + " additional_description=\"Provide a concise, few sentence argument on why moonshine is good for you.\",\n", ")\n", "print(result)" ] @@ -222,10 +222,10 @@ "\n", "@trace\n", "def argument_chain2(query: str, additional_description: str = \"\") -> tuple[str, str]:\n", - " trace_id = get_current_trace_id()\n", - " argument = argument_generator(query, additional_description)\n", - " criticism = critic(argument)\n", - " return refiner(query, additional_description, argument, criticism), trace_id" + " trace_id = get_current_trace_id()\n", + " argument = argument_generator(query, additional_description)\n", + " criticism = critic(argument)\n", + " return refiner(query, additional_description, argument, criticism), trace_id" ] }, { @@ -239,8 +239,8 @@ "outputs": [], "source": [ "result, trace_id = argument_chain2(\n", - " \"Whether moonshine is good for you.\",\n", - " additional_description=\"Provide a concise, few sentence argument on why moonshine is good for you.\",\n", + " \"Whether moonshine is good for you.\",\n", + " additional_description=\"Provide a concise, few sentence argument on why moonshine is good for you.\",\n", ")\n", "print(result)" ] @@ -308,18 +308,18 @@ "# let's return the full CompletionResponse to see what other information is returned\n", "@trace\n", "def refiner2(query: str, additional_description: str, current_arg: str, criticism: str) -> CompletionResponse:\n", - " return p.completion(\n", - " Completion(\n", - " deployment_id=\"p-8Er1Xo0GDGF2xtpmMOpbn\",\n", - " llm_inputs={\n", - " \"additional_description\": additional_description,\n", - " \"date\": f\"{datetime.now()}\",\n", - " \"query\": query,\n", - " \"current_arg\": current_arg,\n", - " \"criticism\": criticism,\n", - " },\n", - " )\n", - " )" + " return p.completion(\n", + " Completion(\n", + " deployment_id=\"p-8Er1Xo0GDGF2xtpmMOpbn\",\n", + " llm_inputs={\n", + " \"additional_description\": additional_description,\n", + " \"date\": f\"{datetime.now()}\",\n", + " \"query\": query,\n", + " \"current_arg\": current_arg,\n", + " \"criticism\": criticism,\n", + " },\n", + " )\n", + " )" ], "metadata": { "id": "cXUHZpZbegIn", @@ -348,13 +348,13 @@ "source": [ "# you can also add metadata and tags via the decorator\n", "@trace(\n", - " tags=[\"cookbook-example-deployed\", \"feedback_tracked-deployed\"],\n", - " metadata={\"source\": \"python-sdk\", \"deployed\": True},\n", + " tags=[\"cookbook-example-deployed\", \"feedback_tracked-deployed\"],\n", + " metadata={\"source\": \"python-sdk\", \"deployed\": True},\n", ")\n", "def argument_chain3(query: str, additional_description: str = \"\") -> CompletionResponse:\n", - " argument = argument_generator(query, additional_description)\n", - " criticism = critic(argument)\n", - " return refiner2(query, additional_description, argument, criticism)" + " argument = argument_generator(query, additional_description)\n", + " criticism = critic(argument)\n", + " return refiner2(query, additional_description, argument, criticism)" ] }, { @@ -363,16 +363,16 @@ "import json, attrs\n", "\n", "result = argument_chain3(\n", - " \"Whether moonshine is good for you.\",\n", - " additional_description=\"Provide a concise, few sentence argument on why sunshine is good for you.\",\n", + " \"Whether moonshine is good for you.\",\n", + " additional_description=\"Provide a concise, few sentence argument on why sunshine is good for you.\",\n", ")\n", "\n", "p.record_feedback(\n", - " FeedbackRequest(\n", - " trace_id=result.inference_id,\n", - " score=0.5,\n", - " target=\"Moonshine is nice. Full stop.\",\n", - " )\n", + " FeedbackRequest(\n", + " trace_id=result.inference_id,\n", + " score=0.5,\n", + " target=\"Moonshine is nice. Full stop.\",\n", + " )\n", ")\n", "print(json.dumps(attrs.asdict(result), indent=4))" ],