feat: implementing initial-setup

app/config/db.py

@@ -1,4 +1,4 @@
-from typing import Literal, Optional
+from typing import Literal
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
@@ -11,7 +11,7 @@ class DBConfig(BaseSettings):
     )
 
 
-    DATABASE_TYPE: str = "mongodb"
+    DATABASE_TYPE: Literal["mongodb", "embedded"] = "embedded"
     DATABASE_NAME: str = "openai_chatbot_api"
 
     MONGO_USER: str = "root"

app/core/initial_setup/data/initial_chat_completions.json

@@ -0,0 +1,124 @@
+{
+    "chat_completions": [
+        {
+            "completion_id": "mock-1",
+            "title": "Customer count by status",
+            "model": "gpt-3.5-turbo",
+            "object": "chat.completion",
+            "is_archived": false,
+            "is_starred": false,
+            "created_by": "system",
+            "created_date": "2024-03-20T10:00:00Z",
+            "last_updated_by": "system",
+            "last_updated_date": "2024-03-20T11:00:00Z",
+            "messages": [
+                {
+                    "message_id": "mock-1-1",
+                    "role": "user",
+                    "content": "count of customers by status",
+                    "created_date": "2024-03-20T10:01:01Z"
+                },
+                {
+                    "message_id": "mock-1-2",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|active|100|\n|cancelled|30|\n|pending|40|\n|passive|150|\n\n## Explanation\nThis result represents the number of active, cancelled, pending and passive customers in the year 2025. This is a mock result for the question \"How many customers are active, cancelled, pending and passive in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM customers WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": null,
+                    "created_date": "2024-03-20T10:02:02Z"
+                },
+                {
+                    "message_id": "mock-1-3",
+                    "role": "user",
+                    "content": "show me the result in a bar chart",
+                    "created_date": "2024-03-20T10:03:03Z"
+                },
+                {
+                    "message_id": "mock-1-4",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|active|100|\n|cancelled|30|\n|pending|40|\n|passive|150|\n\n## Explanation\nThis result represents the number of active, cancelled, pending and passive customers in the year 2025. This is a mock result for the question \"How many customers are active, cancelled, pending and passive in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM customers WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": {"data":[{"alignmentgroup":"True","customdata":[["North"],["North"],["South"]],"hovertemplate":"category=Category A\u003cbr\u003estatus=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003eregion=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"Category A","marker":{"color":"#636efa","pattern":{"shape":""}},"name":"Category A","offsetgroup":"Category A","orientation":"v","showlegend":true,"textposition":"auto","texttemplate":"%{y}","x":["active","in_review","pending_deletion"],"xaxis":"x","y":{"dtype":"i1","bdata":"Mh4P"},"yaxis":"y","type":"bar"},{"alignmentgroup":"True","customdata":[["South"],["West"]],"hovertemplate":"category=Category B\u003cbr\u003estatus=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003eregion=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"Category B","marker":{"color":"#EF553B","pattern":{"shape":""}},"name":"Category B","offsetgroup":"Category B","orientation":"v","showlegend":true,"textposition":"auto","texttemplate":"%{y}","x":["passive","newly_added"],"xaxis":"x","y":{"dtype":"i1","bdata":"Rhk="},"yaxis":"y","type":"bar"},{"alignmentgroup":"True","customdata":[["East"]],"hovertemplate":"category=Category C\u003cbr\u003estatus=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003eregion=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"Category C","marker":{"color":"#00cc96","pattern":{"shape":""}},"name":"Category C","offsetgroup":"Category C","orientation":"v","showlegend":true,"textposition":"auto","texttemplate":"%{y}","x":["archived"],"xaxis":"x","y":{"dtype":"i1","bdata":"Wg=="},"yaxis":"y","type":"bar"}],"layout":{"template":{"data":{"histogram2dcontour":[{"type":"histogram2dcontour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"choropleth":[{"type":"choropleth","colorbar":{"outlinewidth":0,"ticks":""}}],"histogram2d":[{"type":"histogram2d","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"heatmap":[{"type":"heatmap","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"contourcarpet":[{"type":"contourcarpet","colorbar":{"outlinewidth":0,"ticks":""}}],"contour":[{"type":"contour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"surface":[{"type":"surface","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"mesh3d":[{"type":"mesh3d","colorbar":{"outlinewidth":0,"ticks":""}}],"scatter":[{"fillpattern":{"fillmode":"overlay","size":10,"solidity":0.2},"type":"scatter"}],"parcoords":[{"type":"parcoords","line":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolargl":[{"type":"scatterpolargl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"bar":[{"error_x":{"color":"#2a3f5f"},"error_y":{"color":"#2a3f5f"},"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"bar"}],"scattergeo":[{"type":"scattergeo","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolar":[{"type":"scatterpolar","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"histogram":[{"marker":{"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"histogram"}],"scattergl":[{"type":"scattergl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatter3d":[{"type":"scatter3d","line":{"colorbar":{"outlinewidth":0,"ticks":""}},"marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermap":[{"type":"scattermap","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermapbox":[{"type":"scattermapbox","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterternary":[{"type":"scatterternary","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattercarpet":[{"type":"scattercarpet","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"carpet":[{"aaxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"baxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"type":"carpet"}],"table":[{"cells":{"fill":{"color":"#EBF0F8"},"line":{"color":"white"}},"header":{"fill":{"color":"#C8D4E3"},"line":{"color":"white"}},"type":"table"}],"barpolar":[{"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"barpolar"}],"pie":[{"automargin":true,"type":"pie"}]},"layout":{"autotypenumbers":"strict","colorway":["#636efa","#EF553B","#00cc96","#ab63fa","#FFA15A","#19d3f3","#FF6692","#B6E880","#FF97FF","#FECB52"],"font":{"color":"#2a3f5f"},"hovermode":"closest","hoverlabel":{"align":"left"},"paper_bgcolor":"white","plot_bgcolor":"#E5ECF6","polar":{"bgcolor":"#E5ECF6","angularaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"radialaxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"ternary":{"bgcolor":"#E5ECF6","aaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"baxis":{"gridcolor":"white","linecolor":"white","ticks":""},"caxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"coloraxis":{"colorbar":{"outlinewidth":0,"ticks":""}},"colorscale":{"sequential":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"sequentialminus":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"diverging":[[0,"#8e0152"],[0.1,"#c51b7d"],[0.2,"#de77ae"],[0.3,"#f1b6da"],[0.4,"#fde0ef"],[0.5,"#f7f7f7"],[0.6,"#e6f5d0"],[0.7,"#b8e186"],[0.8,"#7fbc41"],[0.9,"#4d9221"],[1,"#276419"]]},"xaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"yaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"scene":{"xaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"yaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"zaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2}},"shapedefaults":{"line":{"color":"#2a3f5f"}},"annotationdefaults":{"arrowcolor":"#2a3f5f","arrowhead":0,"arrowwidth":1},"geo":{"bgcolor":"white","landcolor":"#E5ECF6","subunitcolor":"white","showland":true,"showlakes":true,"lakecolor":"white"},"title":{"x":0.05},"mapbox":{"style":"light"}}},"xaxis":{"anchor":"y","domain":[0.0,1.0],"title":{"text":"Status"}},"yaxis":{"anchor":"x","domain":[0.0,1.0],"title":{"text":"Count"}},"legend":{"title":{"text":"Category"},"tracegroupgap":0},"title":{"text":"Bar Chart: Count by Status and Category","x":0.5},"barmode":"group"}},
+                    "created_date": "2024-03-20T10:04:04Z"
+                }
+            ]
+        },
+        {
+            "completion_id": "mock-2",
+            "title": "Ticket count by status",
+            "model": "gpt-3.5-turbo",
+            "object": "chat.completion",
+            "is_archived": false,
+            "is_starred": false,
+            "created_by": "system",
+            "created_date": "2024-03-21T10:00:00Z",
+            "last_updated_by": "system",
+            "last_updated_date": "2024-03-21T11:00:00Z",
+            "messages": [
+                {
+                    "message_id": "mock-2-1",
+                    "role": "user",
+                    "content": "count of tickets by status",
+                    "created_date": "2024-03-20T11:01:01Z"
+                },
+                {
+                    "message_id": "mock-2-2",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|open|100|\n|closed|30|\n|pending|40|\n|reopened|150|\n\n## Explanation\nThis result represents the number of open, closed, pending and reopened tickets in the year 2025. This is a mock result for the question \"How many tickets are open, closed, pending and reopened in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM tickets WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": null,
+                    "created_date": "2024-03-20T11:02:02Z"
+                },
+                {
+                    "message_id": "mock-2-3",
+                    "role": "user",
+                    "content": "show me the result in a pie chart",
+                    "created_date": "2024-03-20T11:03:03Z"
+                },
+                {
+                    "message_id": "mock-2-4",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|open|100|\n|closed|30|\n|pending|40|\n|reopened|150|\n\n## Explanation\nThis result represents the number of open, closed, pending and reopened tickets in the year 2025. This is a mock result for the question \"How many tickets are open, closed, pending and reopened in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM tickets WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": {"data":[{"customdata":[[150000,"Dr. Eva Core","High"],[120000,"Mr. Alex Markey","High"],[85000,"Ms. Sarah Ops","Medium"],[45000,"Mr. John Staff","Medium"],[65000,"Ms. Ada Byte","High"],[35000,"Mrs. Jane Doe","Low"]],"domain":{"x":[0.0,1.0],"y":[0.0,1.0]},"hole":0.4,"hovertemplate":"\u003cb\u003e%{hovertext}\u003c\u002fb\u003e\u003cbr\u003e\u003cbr\u003edepartment=%{label}\u003cbr\u003ebudget_allocation_usd=%{customdata[0]:$,.2f}\u003cbr\u003elead_person=%{customdata[1]}\u003cbr\u003epriority_level=%{customdata[2]}\u003cextra\u003e\u003c\u002fextra\u003e","hovertext":["Research & Development","Marketing & Sales","Operations","Human Resources","IT Infrastructure","Administration & Support"],"labels":["Research & Development","Marketing & Sales","Operations","Human Resources","IT Infrastructure","Administration & Support"],"legendgroup":"","name":"","showlegend":true,"values":{"dtype":"i4","bdata":"8EkCAMDUAQAITAEAyK8AAOj9AAC4iAAA"},"type":"pie","marker":{"line":{"color":"#000000","width":1.5}},"textinfo":"percent+label","textposition":"inside"}],"layout":{"template":{"data":{"histogram2dcontour":[{"type":"histogram2dcontour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"choropleth":[{"type":"choropleth","colorbar":{"outlinewidth":0,"ticks":""}}],"histogram2d":[{"type":"histogram2d","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"heatmap":[{"type":"heatmap","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"contourcarpet":[{"type":"contourcarpet","colorbar":{"outlinewidth":0,"ticks":""}}],"contour":[{"type":"contour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"surface":[{"type":"surface","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"mesh3d":[{"type":"mesh3d","colorbar":{"outlinewidth":0,"ticks":""}}],"scatter":[{"fillpattern":{"fillmode":"overlay","size":10,"solidity":0.2},"type":"scatter"}],"parcoords":[{"type":"parcoords","line":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolargl":[{"type":"scatterpolargl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"bar":[{"error_x":{"color":"#2a3f5f"},"error_y":{"color":"#2a3f5f"},"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"bar"}],"scattergeo":[{"type":"scattergeo","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolar":[{"type":"scatterpolar","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"histogram":[{"marker":{"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"histogram"}],"scattergl":[{"type":"scattergl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatter3d":[{"type":"scatter3d","line":{"colorbar":{"outlinewidth":0,"ticks":""}},"marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermap":[{"type":"scattermap","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermapbox":[{"type":"scattermapbox","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterternary":[{"type":"scatterternary","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattercarpet":[{"type":"scattercarpet","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"carpet":[{"aaxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"baxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"type":"carpet"}],"table":[{"cells":{"fill":{"color":"#EBF0F8"},"line":{"color":"white"}},"header":{"fill":{"color":"#C8D4E3"},"line":{"color":"white"}},"type":"table"}],"barpolar":[{"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"barpolar"}],"pie":[{"automargin":true,"type":"pie"}]},"layout":{"autotypenumbers":"strict","colorway":["#636efa","#EF553B","#00cc96","#ab63fa","#FFA15A","#19d3f3","#FF6692","#B6E880","#FF97FF","#FECB52"],"font":{"color":"#2a3f5f"},"hovermode":"closest","hoverlabel":{"align":"left"},"paper_bgcolor":"white","plot_bgcolor":"#E5ECF6","polar":{"bgcolor":"#E5ECF6","angularaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"radialaxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"ternary":{"bgcolor":"#E5ECF6","aaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"baxis":{"gridcolor":"white","linecolor":"white","ticks":""},"caxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"coloraxis":{"colorbar":{"outlinewidth":0,"ticks":""}},"colorscale":{"sequential":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"sequentialminus":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"diverging":[[0,"#8e0152"],[0.1,"#c51b7d"],[0.2,"#de77ae"],[0.3,"#f1b6da"],[0.4,"#fde0ef"],[0.5,"#f7f7f7"],[0.6,"#e6f5d0"],[0.7,"#b8e186"],[0.8,"#7fbc41"],[0.9,"#4d9221"],[1,"#276419"]]},"xaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"yaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"scene":{"xaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"yaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"zaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2}},"shapedefaults":{"line":{"color":"#2a3f5f"}},"annotationdefaults":{"arrowcolor":"#2a3f5f","arrowhead":0,"arrowwidth":1},"geo":{"bgcolor":"white","landcolor":"#E5ECF6","subunitcolor":"white","showland":true,"showlakes":true,"lakecolor":"white"},"title":{"x":0.05},"mapbox":{"style":"light"}}},"legend":{"tracegroupgap":0,"title":{"text":"Departments"},"font":{"size":11},"orientation":"v","yanchor":"top","y":0.85,"xanchor":"left","x":1.02,"traceorder":"reversed","bgcolor":"rgba(255,255,255,0.7)","bordercolor":"Gray","borderwidth":1},"title":{"text":"\u003cb\u003eProject Budget Distribution by Department\u003c\u002fb\u003e","font":{"size":22,"family":"Arial Black, sans-serif","color":"#333333"},"y":0.95,"x":0.5,"xanchor":"center","yanchor":"top"},"font":{"family":"Arial, sans-serif","size":12,"color":"#444444"},"paper_bgcolor":"rgb(248, 248, 255)","plot_bgcolor":"rgba(0,0,0,0)","annotations":[{"showarrow":false,"text":"Total Budget\u003cbr\u003e$400,000","x":0.5,"y":0.5,"font":{"size":16,"color":"#1f77b4"}}]}},
+                    "created_date": "2024-03-20T11:04:04Z"
+                }
+            ]
+        },
+        {
+            "completion_id": "mock-3",
+            "title": "Ticket comments count by customer status",
+            "model": "gpt-3.5-turbo",
+            "object": "chat.completion",
+            "is_archived": false,
+            "is_starred": false,
+            "created_by": "system",
+            "created_date": "2024-03-22T10:00:00Z",
+            "last_updated_by": "system",
+            "last_updated_date": "2024-03-22T11:00:00Z",
+            "messages": [
+                {
+                    "message_id": "mock-3-1",
+                    "role": "user",
+                    "content": "count of ticket comments by status",
+                    "created_date": "2024-03-22T11:01:01Z"
+                },
+                {
+                    "message_id": "mock-3-2",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|open|100|\n|closed|30|\n|pending|40|\n|reopened|150|\n\n## Explanation\nThis result represents the number of open, closed, pending and reopened tickets in the year 2025. This is a mock result for the question \"How many tickets are open, closed, pending and reopened in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM tickets WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": null,
+                    "created_date": "2024-03-22T11:02:02Z"
+                },
+                {
+                    "message_id": "mock-3-3",
+                    "role": "user",
+                    "content": "show me the result in a line chart",
+                    "created_date": "2024-03-22T11:03:03Z"
+                },
+                {
+                    "message_id": "mock-3-4",
+                    "role": "assistant",
+                    "content": "## Result\n|status|count|\n|---|---|\n|open|100|\n|closed|30|\n|pending|40|\n|reopened|150|\n\n## Explanation\nThis result represents the number of open, closed, pending and reopened tickets in the year 2025. This is a mock result for the question \"How many tickets are open, closed, pending and reopened in the year 2025?\"\n\n## Query\n```sql\nSELECT status, COUNT(*) FROM tickets WHERE year = 2025 GROUP BY status;\n```",
+                    "figure": {"data":[{"customdata":[["math"],["math"],["science"],["science"],["art"],["art"],["science"],["science"],["math"]],"hovertemplate":"status=active\u003cbr\u003edate=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003esubject=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"active","line":{"color":"#3498db","dash":"solid"},"marker":{"symbol":"circle"},"mode":"lines+markers","name":"active","orientation":"v","showlegend":true,"x":["2025-01-01T00:00:00.000000000","2025-02-01T00:00:00.000000000","2025-03-01T00:00:00.000000000","2025-04-01T00:00:00.000000000","2025-05-01T00:00:00.000000000","2025-06-01T00:00:00.000000000","2025-07-01T00:00:00.000000000","2025-08-01T00:00:00.000000000","2025-09-01T00:00:00.000000000"],"xaxis":"x","y":{"dtype":"i1","bdata":"Cg8MEg4UEBYZ"},"yaxis":"y","type":"scatter"},{"customdata":[["math"],["math"],["science"],["science"],["math"],["math"],["music"],["music"],["history"]],"hovertemplate":"status=archived\u003cbr\u003edate=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003esubject=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"archived","line":{"color":"#9b59b6","dash":"solid"},"marker":{"symbol":"circle"},"mode":"lines+markers","name":"archived","orientation":"v","showlegend":true,"x":["2025-01-01T00:00:00.000000000","2025-02-01T00:00:00.000000000","2025-03-01T00:00:00.000000000","2025-04-01T00:00:00.000000000","2025-05-01T00:00:00.000000000","2025-06-01T00:00:00.000000000","2025-07-01T00:00:00.000000000","2025-08-01T00:00:00.000000000","2025-09-01T00:00:00.000000000"],"xaxis":"x","y":{"dtype":"i1","bdata":"KC0qMCwyLzc8"},"yaxis":"y","type":"scatter"},{"customdata":[["english"],["english"],["math"],["math"],["history"],["history"],["art"],["art"],["english"]],"hovertemplate":"status=in_review\u003cbr\u003edate=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003esubject=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"in_review","line":{"color":"#2ecc71","dash":"solid"},"marker":{"symbol":"circle"},"mode":"lines+markers","name":"in_review","orientation":"v","showlegend":true,"x":["2025-01-01T00:00:00.000000000","2025-02-01T00:00:00.000000000","2025-03-01T00:00:00.000000000","2025-04-01T00:00:00.000000000","2025-05-01T00:00:00.000000000","2025-06-01T00:00:00.000000000","2025-07-01T00:00:00.000000000","2025-08-01T00:00:00.000000000","2025-09-01T00:00:00.000000000"],"xaxis":"x","y":{"dtype":"i1","bdata":"HiMgJh4kIScq"},"yaxis":"y","type":"scatter"},{"customdata":[["science"],["science"],["english"],["english"],["music"],["music"],["english"],["english"],["science"]],"hovertemplate":"status=passive\u003cbr\u003edate=%{x}\u003cbr\u003ecount=%{y}\u003cbr\u003esubject=%{customdata[0]}\u003cextra\u003e\u003c\u002fextra\u003e","legendgroup":"passive","line":{"color":"#e74c3c","dash":"solid"},"marker":{"symbol":"circle"},"mode":"lines+markers","name":"passive","orientation":"v","showlegend":true,"x":["2025-01-01T00:00:00.000000000","2025-02-01T00:00:00.000000000","2025-03-01T00:00:00.000000000","2025-04-01T00:00:00.000000000","2025-05-01T00:00:00.000000000","2025-06-01T00:00:00.000000000","2025-07-01T00:00:00.000000000","2025-08-01T00:00:00.000000000","2025-09-01T00:00:00.000000000"],"xaxis":"x","y":{"dtype":"i1","bdata":"FBkWHBQaEhgP"},"yaxis":"y","type":"scatter"}],"layout":{"template":{"data":{"histogram2dcontour":[{"type":"histogram2dcontour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"choropleth":[{"type":"choropleth","colorbar":{"outlinewidth":0,"ticks":""}}],"histogram2d":[{"type":"histogram2d","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"heatmap":[{"type":"heatmap","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"contourcarpet":[{"type":"contourcarpet","colorbar":{"outlinewidth":0,"ticks":""}}],"contour":[{"type":"contour","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"surface":[{"type":"surface","colorbar":{"outlinewidth":0,"ticks":""},"colorscale":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]]}],"mesh3d":[{"type":"mesh3d","colorbar":{"outlinewidth":0,"ticks":""}}],"scatter":[{"fillpattern":{"fillmode":"overlay","size":10,"solidity":0.2},"type":"scatter"}],"parcoords":[{"type":"parcoords","line":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolargl":[{"type":"scatterpolargl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"bar":[{"error_x":{"color":"#2a3f5f"},"error_y":{"color":"#2a3f5f"},"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"bar"}],"scattergeo":[{"type":"scattergeo","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterpolar":[{"type":"scatterpolar","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"histogram":[{"marker":{"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"histogram"}],"scattergl":[{"type":"scattergl","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatter3d":[{"type":"scatter3d","line":{"colorbar":{"outlinewidth":0,"ticks":""}},"marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermap":[{"type":"scattermap","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattermapbox":[{"type":"scattermapbox","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scatterternary":[{"type":"scatterternary","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"scattercarpet":[{"type":"scattercarpet","marker":{"colorbar":{"outlinewidth":0,"ticks":""}}}],"carpet":[{"aaxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"baxis":{"endlinecolor":"#2a3f5f","gridcolor":"white","linecolor":"white","minorgridcolor":"white","startlinecolor":"#2a3f5f"},"type":"carpet"}],"table":[{"cells":{"fill":{"color":"#EBF0F8"},"line":{"color":"white"}},"header":{"fill":{"color":"#C8D4E3"},"line":{"color":"white"}},"type":"table"}],"barpolar":[{"marker":{"line":{"color":"#E5ECF6","width":0.5},"pattern":{"fillmode":"overlay","size":10,"solidity":0.2}},"type":"barpolar"}],"pie":[{"automargin":true,"type":"pie"}]},"layout":{"autotypenumbers":"strict","colorway":["#636efa","#EF553B","#00cc96","#ab63fa","#FFA15A","#19d3f3","#FF6692","#B6E880","#FF97FF","#FECB52"],"font":{"color":"#2a3f5f"},"hovermode":"closest","hoverlabel":{"align":"left"},"paper_bgcolor":"white","plot_bgcolor":"#E5ECF6","polar":{"bgcolor":"#E5ECF6","angularaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"radialaxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"ternary":{"bgcolor":"#E5ECF6","aaxis":{"gridcolor":"white","linecolor":"white","ticks":""},"baxis":{"gridcolor":"white","linecolor":"white","ticks":""},"caxis":{"gridcolor":"white","linecolor":"white","ticks":""}},"coloraxis":{"colorbar":{"outlinewidth":0,"ticks":""}},"colorscale":{"sequential":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"sequentialminus":[[0.0,"#0d0887"],[0.1111111111111111,"#46039f"],[0.2222222222222222,"#7201a8"],[0.3333333333333333,"#9c179e"],[0.4444444444444444,"#bd3786"],[0.5555555555555556,"#d8576b"],[0.6666666666666666,"#ed7953"],[0.7777777777777778,"#fb9f3a"],[0.8888888888888888,"#fdca26"],[1.0,"#f0f921"]],"diverging":[[0,"#8e0152"],[0.1,"#c51b7d"],[0.2,"#de77ae"],[0.3,"#f1b6da"],[0.4,"#fde0ef"],[0.5,"#f7f7f7"],[0.6,"#e6f5d0"],[0.7,"#b8e186"],[0.8,"#7fbc41"],[0.9,"#4d9221"],[1,"#276419"]]},"xaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"yaxis":{"gridcolor":"white","linecolor":"white","ticks":"","title":{"standoff":15},"zerolinecolor":"white","automargin":true,"zerolinewidth":2},"scene":{"xaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"yaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2},"zaxis":{"backgroundcolor":"#E5ECF6","gridcolor":"white","linecolor":"white","showbackground":true,"ticks":"","zerolinecolor":"white","gridwidth":2}},"shapedefaults":{"line":{"color":"#2a3f5f"}},"annotationdefaults":{"arrowcolor":"#2a3f5f","arrowhead":0,"arrowwidth":1},"geo":{"bgcolor":"white","landcolor":"#E5ECF6","subunitcolor":"white","showland":true,"showlakes":true,"lakecolor":"white"},"title":{"x":0.05},"mapbox":{"style":"light"}}},"xaxis":{"anchor":"y","domain":[0.0,1.0],"title":{"text":"Date"},"showgrid":true,"gridcolor":"#e0e0e0","zeroline":false,"tickformat":"%Y-%m-%d"},"yaxis":{"anchor":"x","domain":[0.0,1.0],"title":{"text":"Count"},"showgrid":true,"gridcolor":"#e0e0e0","zeroline":false},"legend":{"title":{"text":"Status"},"tracegroupgap":0,"orientation":"h","yanchor":"bottom","y":1.02,"xanchor":"right","x":1},"title":{"text":"Line Chart of Count Over Time by Status","font":{"size":20,"family":"Arial","color":"#2c3e50"},"y":0.95,"x":0.5,"xanchor":"center","yanchor":"top"},"font":{"family":"Arial","size":12,"color":"#333"},"plot_bgcolor":"#f5f6f5","paper_bgcolor":"#ffffff","showlegend":true}},
+                    "created_date": "2024-03-22T11:04:04Z"
+                }
+            ]
+        }
+    ]
+}

app/core/initial_setup/setup.py

@@ -0,0 +1,55 @@
+import json
+import os
+from typing import List, Optional
+from loguru import logger
+from app.model.chat_model import ChatCompletion
+from app.repository.chat_repository import ChatRepository
+from app.config.db import db_config
+
+class InitialSetup:
+    """Initial setup manager for the application"""
+    
+    def __init__(self):
+        self._chat_repository: Optional[ChatRepository] = None
+        self.data_dir = os.path.join(os.path.dirname(__file__), "data")
+    
+    @property
+    def chat_repository(self) -> ChatRepository:
+        """Lazy loading of ChatRepository"""
+        if self._chat_repository is None:
+            self._chat_repository = ChatRepository()
+        return self._chat_repository
+    
+    def _load_initial_data(self) -> List[ChatCompletion]:
+        """Load initial data from JSON files"""
+        try:
+            with open(os.path.join(self.data_dir, "initial_chat_completions.json"), "r") as f:
+                data = json.load(f)
+                return [ChatCompletion(**item) for item in data["chat_completions"]]
+        except Exception as e:
+            logger.error(f"Error loading initial data: {e}")
+            return []
+    
+    async def setup(self) -> None:
+        """Setup initial data if database type is embedded"""
+        if db_config.DATABASE_TYPE != "embedded":
+            logger.info("Skipping initial setup as database type is not embedded")
+            return
+        
+        logger.info("Starting initial setup for embedded database")
+        
+        # Load and save initial chat completions
+        chat_completions = self._load_initial_data()
+        for completion in chat_completions:
+            try:
+                self.chat_repository.save(completion)
+                logger.info(f"Saved initial chat completion: {completion.completion_id}")
+            except Exception as e:
+                logger.error(f"Error saving chat completion {completion.completion_id}: {e}")
+        
+        # query of the saved chat completions
+        saved_chat_completions = self.chat_repository.find()
+        logger.info(f"Saved chat completions: {saved_chat_completions}")
+
+
+        logger.info("Initial setup completed successfully") 

app/model/chat_model.py

@@ -5,7 +5,7 @@ from enum import Enum
 from pydantic import BaseModel
 from datetime import datetime
 from pydantic import Field
-from typing import List
+from typing import List, Optional
 
 # Chat completion payload example
 # {
@@ -90,11 +90,7 @@ class ChatCompletion(BaseModel):
     A chat completion.
     """
 
-    completion_id: str = Field(
-        ...,
-        description="The unique identifier for the chat completion",
-        default_factory=lambda: str(uuid.uuid4()),
-    )
+    completion_id: Optional[str] = Field(None, description="The unique identifier for the chat completion")
 
     # openai compatible fields
     model: str = Field(

app/repository/chat_repository.py

@@ -1,10 +1,13 @@
 from typing import List
 from app.db.factory import db_client
 from app.model.chat_model import ChatMessage, ChatCompletion
+from loguru import logger
+import uuid
 
 
 class ChatRepository:
     def __init__(self):
+        logger.info("Initializing ChatRepository")
         self.db = db_client.db
         self.collection = "chat_completion"
 
@@ -12,38 +15,53 @@ class ChatRepository:
         """
         Upsert a chat completion into the database. If the chat completion already exists, it will be updated. If it does not exist, it will be created.
         """
+        logger.debug(f"BEGIN: save chat completion. input data: {entity.model_dump()}")
+        if entity.completion_id is None:
+            logger.warning("completion_id is None. Generating a new one")
+        else:
+            logger.debug(f"completion_id is not None. Using the existing one: {entity.completion_id}")
+
         query = {"completion_id": entity.completion_id}
         update = {"$set": entity.model_dump()}
 
-        result = self.db.chat_completion.update_one(query, update, upsert=True)
+        upserted_entity = self.db.chat_completion.update_one(query, update, upsert=True)
+        logger.debug(f"upserted_entity: {upserted_entity}")
 
-        completion_id = str(result.upserted_id) if result.upserted_id else entity.completion_id
+        completion_id = entity.completion_id if entity.completion_id else str(uuid.uuid4())
         entity.completion_id = completion_id
 
         # save conversation if new chat completion
         # TODO: save conversation
 
-        return entity
+        result = self.find_by_id(completion_id)
+        logger.debug("END: save chat completion")
+        return result
 
-    def find(self, query: dict, page: int = 1, limit: int = 10, sort: dict = None, project: dict = None) -> List[ChatCompletion]:
+    def find(self, query: dict = {}, page: int = 1, limit: int = 10, sort: dict = {"created_date": -1}, project: dict = None) -> List[ChatCompletion]:
         """
         Find a chat completion by a given query. with pagination
         Example : query = {"created_by": "admin"}
         """
-
+        logger.debug(f"BEGIN: find chat completion. input parameters: query: {query}, page: {page}, limit: {limit}, sort: {sort}, project: {project}")
         skip = (page - 1) * limit
         sort = sort if sort else {"created_date": -1}
 
         result = self.db.chat_completion.find(query, project).skip(skip).limit(limit).sort(sort)
-        return [ChatCompletion(**item) for item in result]
+        logger.debug(f"result: {result}")
+        result = [ChatCompletion(**item) for item in result]
+        logger.debug("END: find chat completion")
+        return result
 
     def find_by_id(self, completion_id: str, project: dict = None) -> ChatCompletion:
         """
         Find a chat completion by a given id.
         Example : completion_id = "123"
         """
-        result = self.db.chat_completion.find_one({"completion_id": completion_id}, project)
-        return ChatCompletion(**result)
+        logger.debug(f"BEGIN: find chat completion by id. input parameters: completion_id: {completion_id}, project: {project}")
+        entity = self.db.chat_completion.find_one({"completion_id": completion_id}, project)
+        result = ChatCompletion(**entity)
+        logger.debug("END: find chat completion by id")
+        return result
 
     def find_messages(self, completion_id: str) -> List[ChatMessage]:
         """

main.py

@@ -12,6 +12,7 @@ from app.db.factory import db_client
 from gradio_chatbot import build_gradio_app, app_auth
 import gradio as gr
 import os
+from app.core.initial_setup.setup import InitialSetup
 
 print(log_config.get_log_level())
 
@@ -34,6 +35,11 @@ async def lifespan(app: FastAPI):
     # Startup
     logger.info("Starting up application...")
     await db_client.connect()
+    
+    # Run initial setup if database type is embedded
+    initial_setup = InitialSetup()
+    await initial_setup.setup()
+    
     yield
 
     # Shutdown