Skip to content

Commit

Permalink
codestyle
Browse files Browse the repository at this point in the history
  • Loading branch information
@jalexanderII
jalexanderII committed Sep 9, 2023
1 parent bb59530 commit b80148f
Showing 1 changed file with 64 additions and 64 deletions.
128 changes: 64 additions & 64 deletions parea/cookbook/tracing_with_Parea_sdk.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -112,47 +112,47 @@
"\n",
"\n",
"def argument_generator(query: str, additional_description: str = \"\") -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-XOh3kp8B0nIE82WgioPnr\",\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-XOh3kp8B0nIE82WgioPnr\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" },\n",
" )\n",
" ).content\n",
"\n",
"\n",
"def critic(argument: str) -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-PSOwRyIPaQRq4xQW3MbpV\",\n",
" llm_inputs={\"argument\": argument},\n",
" )\n",
" ).content\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-PSOwRyIPaQRq4xQW3MbpV\",\n",
" llm_inputs={\"argument\": argument},\n",
" )\n",
" ).content\n",
"\n",
"\n",
"def refiner(query: str, additional_description: str, argument: str, criticism: str) -> str:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-bJ3-UKh9-ixapZafaRBsj\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": argument,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" ).content\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-bJ3-UKh9-ixapZafaRBsj\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": argument,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" ).content\n",
"\n",
"\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 @@ -180,8 +180,8 @@
"outputs": [],
"source": [
"result = argument_chain(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(result)"
]
Expand Down Expand Up @@ -218,10 +218,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 @@ -235,8 +235,8 @@
"outputs": [],
"source": [
"result, trace_id = argument_chain2(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(result)"
]
Expand Down Expand Up @@ -286,18 +286,18 @@
"\n",
"# let's return the full CompletionResponse to see what other information is returned\n",
"def refiner2(query: str, additional_description: str, argument: str, criticism: str) -> CompletionResponse:\n",
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-bJ3-UKh9-ixapZafaRBsj\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": argument,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" )"
" return p.completion(\n",
" Completion(\n",
" deployment_id=\"p-bJ3-UKh9-ixapZafaRBsj\",\n",
" llm_inputs={\n",
" \"additional_description\": additional_description,\n",
" \"date\": f\"{datetime.now()}\",\n",
" \"query\": query,\n",
" \"argument\": argument,\n",
" \"criticism\": criticism,\n",
" },\n",
" )\n",
" )"
],
"metadata": {
"collapsed": false
Expand Down Expand Up @@ -342,14 +342,14 @@
"\n",
"# 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_chain_tags_metadata(query: str, additional_description: str = \"\") -> Tuple[CompletionResponse, str]:\n",
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism), trace_id"
" trace_id = get_current_trace_id() # get parent's trace_id\n",
" argument = argument_generator(query, additional_description)\n",
" criticism = critic(argument)\n",
" return refiner2(query, additional_description, argument, criticism), trace_id"
]
},
{
Expand All @@ -358,17 +358,17 @@
"import json, attrs\n",
"\n",
"result2, trace_id = argument_chain_tags_metadata(\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
" \"Whether coffee is good for you.\",\n",
" additional_description=\"Provide a concise, few sentence argument on why coffee is good for you.\",\n",
")\n",
"print(json.dumps(attrs.asdict(result2), indent=2))\n",
"\n",
"p.record_feedback(\n",
" FeedbackRequest(\n",
" trace_id=trace_id,\n",
" score=0.7, # 0.0 (bad) to 1.0 (good)\n",
" target=\"Coffee is wonderful. End of story.\",\n",
" )\n",
" FeedbackRequest(\n",
" trace_id=trace_id,\n",
" score=0.7, # 0.0 (bad) to 1.0 (good)\n",
" target=\"Coffee is wonderful. End of story.\",\n",
" )\n",
")"
],
"metadata": {
Expand Down

0 comments on commit b80148f

Please sign in to comment.