Merge pull request #30 from beautiful-code/til_v2

endpoints.py

@@ -1,18 +1,18 @@
 from dotenv import load_dotenv
 load_dotenv()
 
-from .workflows.courses.expectation_revision import ExpectationRevision, Inputs as ExpectationRevisionInputs, Response as ExpectationRevisionResponse
-from .workflows.courses.suggest_check_question import SuggestCheckQuestion, Inputs as SuggestCheckQuestionInputs, Response as SuggestCheckQuestionResponse
-from .workflows.courses.suggest_expectations import SuggestExpectations, Inputs as SuggestExpectationsInputs, Expectation, Response as SuggestExpectationsResponse
-from .workflows.til.analyse_til import TilCrew, TilFeedbackResponse
-from .workflows.til.analyse_til_v2 import AnalyseTilV2, TilV2FeedbackResponse
-from .workflows.utils.feedback import Feedback, post_feedback
-from fastapi import FastAPI, Query
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import UUID4
-from tempenv import TemporaryEnvironment
-from typing import List
 import uvicorn
+from typing import List
+from tempenv import TemporaryEnvironment
+from pydantic import UUID4
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, Query
+from .workflows.utils.feedback import Feedback, post_feedback
+from .workflows.til.analyse_til_v2 import AnalyseTilV2, TilV2FeedbackResponse
+from .workflows.til.analyse_til import TilCrew, TilFeedbackResponse
+from .workflows.courses.suggest_expectations import SuggestExpectations, Inputs as SuggestExpectationsInputs, Expectation, Response as SuggestExpectationsResponse
+from .workflows.courses.suggest_check_question import SuggestCheckQuestion, Inputs as SuggestCheckQuestionInputs, Response as SuggestCheckQuestionResponse
+from .workflows.courses.expectation_revision import ExpectationRevision, Inputs as ExpectationRevisionInputs, Response as ExpectationRevisionResponse
 
 
 LANGSMITH_STAGING_PROJECT = "customer_agent"
@@ -108,6 +108,22 @@ async def staging_til_v2_feedback_kickoff(content: List[str]) -> TilV2FeedbackRe
         return til_v2_logic(content)
 
 
+@app.post("/v2/til_feedback/{run_id}/feedback", tags=["til_feedback"])
+async def capture_feedback(run_id: UUID4, feedback: Feedback) -> str:
+    print("Metric Type: ", feedback.metric_type)
+    print("Feedback On: ", feedback.feedback_on)
+    post_feedback(run_id=run_id, feedback=feedback)
+    return "ok"
+
+
+@app.post("/staging/v2/til_feedback/{run_id}/feedback", tags=["til_feedback", "staging"])
+async def capture_feedback(run_id: UUID4, feedback: Feedback) -> str:
+    print("Metric Type: ", feedback.metric_type)
+    print("Feedback On: ", feedback.feedback_on)
+    post_feedback(run_id=run_id, feedback=feedback)
+    return "ok"
+
+
 def course_learn_suggest_expectations_logic(inputs) -> SuggestExpectationsResponse:
     print("Inputs: ", inputs)
     result = SuggestExpectations().kickoff(inputs={