Skip to content

Commit

Permalink
codestyle
Browse files Browse the repository at this point in the history
  • Loading branch information
@jalexanderII
jalexanderII committed Aug 29, 2023
1 parent 5196212 commit 76aacaf
Showing 1 changed file with 63 additions and 63 deletions.
126 changes: 63 additions & 63 deletions parea/cookbook/tracing_with_Parea_sdk.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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)"
]
},
{
Expand Down Expand Up @@ -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)"
]
Expand Down Expand Up @@ -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"
]
},
{
Expand All @@ -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)"
]
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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)"
]
},
{
Expand All @@ -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))"
],
Expand Down

0 comments on commit 76aacaf

Please sign in to comment.