divi 0.0.1.dev13__tar.gz → 0.0.1.dev14__tar.gz

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/.github/dependabot.yml

@@ -5,7 +5,7 @@
 
 version: 2
 updates:
-  - package-ecosystem: "" # See documentation for possible values
+  - package-ecosystem: uv # See documentation for possible values
     directory: "/" # Location of package manifests
     schedule:
       interval: "weekly"

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: divi
-Version: 0.0.1.dev13
+Version: 0.0.1.dev14
 Summary: The Agent Platform for Observability & Evaluation
 License-File: LICENSE
 Requires-Python: >=3.11

divi-0.0.1.dev14/README.md

@@ -0,0 +1,14 @@
+# divine-agent
+
+**Agent Platform for Observability**
+
+## TODO
+
+- [ ] Web
+- [ ] Docs
+
+## Architecture
+
+![architecture](./docs/images/architecture.svg)
+
+![storage](./docs/images/storage.svg)

divi-0.0.1.dev14/apps/docs/README.md

@@ -0,0 +1,32 @@
+# Mintlify Starter Kit
+
+Click on `Use this template` to copy the Mintlify starter kit. The starter kit contains examples including
+
+- Guide pages
+- Navigation
+- Customizations
+- API Reference pages
+- Use of popular components
+
+### Development
+
+Install the [Mintlify CLI](https://www.npmjs.com/package/mintlify) to preview the documentation changes locally. To install, use the following command
+
+```
+npm i -g mintlify
+```
+
+Run the following command at the root of your documentation (where docs.json is)
+
+```
+mintlify dev
+```
+
+### Publishing Changes
+
+Install our Github App to auto propagate changes from your repo to your deployment. Changes will be deployed to production automatically after pushing to the default branch. Find the link to install on your dashboard. 
+
+#### Troubleshooting
+
+- Mintlify dev isn't running - Run `mintlify install` it'll re-install dependencies.
+- Page loads as a 404 - Make sure you are running in a folder with `docs.json`

divi-0.0.1.dev14/apps/docs/concepts/session.mdx

@@ -0,0 +1,4 @@
+---
+title: 'Session'
+description: 'Session is a one-time run of your codes'
+---

divi-0.0.1.dev14/apps/docs/development/roadmap.mdx

@@ -0,0 +1,4 @@
+---
+title: 'Roadmap'
+description: 'Development Roadmap for Divine Agent'
+---

divi-0.0.1.dev14/apps/docs/docs.json

@@ -0,0 +1,68 @@
+{
+  "$schema": "https://mintlify.com/docs.json",
+  "theme": "maple",
+  "name": "Divine Agent",
+  "colors": {
+    "primary": "#09090b",
+    "light": "#FAFAFA",
+    "dark": "#09090b"
+  },
+  "favicon": "/favicon.ico",
+  "navigation": {
+    "groups": [
+      {
+        "group": "Get Started",
+        "pages": ["introduction"]
+      },
+      {
+        "group": "Tutorials",
+        "pages": ["tutorials/openai"]
+      },
+      {
+        "group": "Concepts",
+        "pages": ["concepts/session"]
+      },
+      {
+        "group": "Development",
+        "pages": ["development/roadmap"]
+      }
+    ],
+    "global": {
+      "anchors": [
+        {
+          "anchor": "Dashboard",
+          "href": "https://divine-agent.com",
+          "icon": "gauge-high"
+        }
+      ]
+    }
+  },
+  "logo": {
+    "light": "/images/researcher-angel.png",
+    "dark": "/images/scholar-angel.png"
+  },
+  "navbar": {
+    "links": [
+      {
+        "label": "GitHub",
+        "href": "https://github.com/Kaikaikaifang/divine-agent"
+      }
+    ],
+    "primary": {
+      "type": "button",
+      "label": "Divine Agent",
+      "href": "https://divine-agent.com"
+    }
+  },
+  "seo": {
+    "metatags": {
+      "og:image": "https://raw.githubusercontent.com/Kaikaikaifang/divine-agent/refs/heads/main/docs/images/og-image.png?token=GHSAT0AAAAAAC7SUUQIZIJIKDD5UWUKWX2CZ66TXVA"
+    },
+    "indexing": "navigable"
+  },
+  "footer": {
+    "socials": {
+      "github": "https://github.com/Kaikaikaifang/divine-agent"
+    }
+  }
+}

divi-0.0.1.dev14/apps/docs/introduction.mdx

@@ -0,0 +1,15 @@
+---
+title: Introduction
+description: "Welcome to the docs of Divine Agent"
+---
+
+<img
+  className="block dark:hidden w-20"
+  src="/images/researcher-angel.png"
+  alt="Hero Light"
+/>
+<img
+  className="hidden dark:block w-20"
+  src="/images/scholar-angel.png"
+  alt="Hero Dark"
+/>

divi-0.0.1.dev14/apps/docs/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "@app/docs",
+  "private": true,
+  "scripts": {
+    "dev": "mintlify dev --port 4002 --no-open",
+    "lint": "mintlify broken-links"
+  },
+  "devDependencies": {
+    "typescript": "catalog:"
+  }
+}

divi-0.0.1.dev14/apps/docs/quickstart/trace.mdx

@@ -0,0 +1,4 @@
+---
+title: 'Trace'
+description: 'Start observe your agents in under 5 minutes'
+---

divi-0.0.1.dev14/apps/docs/snippets/snippet-intro.mdx

@@ -0,0 +1,4 @@
+One of the core principles of software development is DRY (Don't Repeat
+Yourself). This is a principle that apply to documentation as
+well. If you find yourself repeating the same content in multiple places, you
+should consider creating a custom snippet to keep your content in sync.

divi-0.0.1.dev14/apps/docs/tutorials/openai.mdx

@@ -0,0 +1,4 @@
+---
+title: 'OpenAI'
+description: 'Start observe your agents with OpenAI SDK'
+---

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/apps/graphql/package.json

@@ -10,19 +10,19 @@
     "postbuild": "cp src/schema.graphql dist/"
   },
   "dependencies": {
-    "graphql": "^16.10.0",
-    "graphql-tag": "^2.12.6",
-    "@apollo/server": "^4.11.3",
     "@apollo/datasource-rest": "^6.4.1",
-    "@apollo/utils.keyvaluecache": "^3.1.0"
+    "@apollo/server": "^4.11.3",
+    "@apollo/utils.keyvaluecache": "^3.1.0",
+    "graphql": "^16.10.0",
+    "graphql-tag": "^2.12.6"
   },
   "devDependencies": {
-    "typescript": "catalog:",
+    "@graphql-codegen/cli": "^4.0.1",
+    "@graphql-codegen/typescript": "^4.1.6",
+    "@graphql-codegen/typescript-resolvers": "^4.5.0",
+    "@workspace/typescript-config": "workspace:*",
     "nodemon": "^3.1.9",
     "ts-node-dev": "^2.0.0",
-    "@graphql-codegen/cli": "^4.0.1",
-    "@graphql-codegen/typescript": "^4.1.5",
-    "@graphql-codegen/typescript-resolvers": "^4.4.4",
-    "@workspace/typescript-config": "workspace:*"
+    "typescript": "catalog:"
   }
 }

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/apps/graphql/src/datasources/auth-api.ts

@@ -1,6 +1,5 @@
-import type { APIKeyModel } from '@/types/api-key';
 import type { FetchResponse } from '@/types/response';
-import type { UserModel } from '@/types/user';
+import type { ApiKey, User } from '@/types/types';
 import { type AugmentedRequest, RESTDataSource } from '@apollo/datasource-rest';
 import type { KeyValueCache } from '@apollo/utils.keyvaluecache';
 
@@ -18,17 +17,21 @@ export class AuthAPI extends RESTDataSource {
   }
 
   async createUser(email: string, password: string, username: string) {
-    return await this.post<FetchResponse<UserModel>>('/api/user/', {
+    return await this.post<FetchResponse<User>>('/api/user/', {
       body: { email, password, username },
     });
   }
 
+  async getCurrentUser() {
+    return await this.get<FetchResponse<User>>('/api/user/');
+  }
+
   async getUser(userId: string) {
-    return await this.get<FetchResponse<UserModel>>(`/api/user/${userId}`);
+    return await this.get<FetchResponse<User>>(`/api/user/${userId}`);
   }
 
-  async updateUser(userId: string, name: string) {
-    return await this.patch<FetchResponse<UserModel>>(`/api/user/${userId}`, {
+  async updateUser(userId: string, name: string | undefined) {
+    return await this.patch<FetchResponse<User>>(`/api/user/${userId}`, {
       body: { name },
     });
   }
@@ -40,11 +43,19 @@ export class AuthAPI extends RESTDataSource {
   }
 
   async getAPIKeys() {
-    return await this.get<FetchResponse<APIKeyModel[]>>('/api/api_key/');
+    return await this.get<FetchResponse<ApiKey[]>>('/api/api_key/');
+  }
+
+  async createAPIKey(name: string) {
+    return await this.post<FetchResponse<ApiKey>>('/api/api_key/', {
+      body: { name },
+    });
   }
 
-  async createAPIKey() {
-    return await this.post<FetchResponse<APIKeyModel>>('/api/api_key/');
+  async updateAPIKey(apiKeyId: string, name: string | undefined) {
+    return await this.patch<FetchResponse<ApiKey>>(`/api/api_key/${apiKeyId}`, {
+      body: { name },
+    });
   }
 
   async revokeAPIKey(apiKeyId: string) {

divi-0.0.1.dev14/apps/graphql/src/datasources/datapark-api.ts

@@ -0,0 +1,32 @@
+import type { FetchResponse } from '@/types/response';
+import type { Span, Trace } from '@/types/types';
+import { type AugmentedRequest, RESTDataSource } from '@apollo/datasource-rest';
+import type { KeyValueCache } from '@apollo/utils.keyvaluecache';
+
+export class DataParkAPI extends RESTDataSource {
+  override baseURL = process.env.AUTH_SERVICE_URL ?? 'http://localhost:3001/';
+  private token: string;
+
+  constructor(options: { token: string; cache: KeyValueCache }) {
+    super(options);
+    this.token = options.token;
+  }
+
+  override willSendRequest(_path: string, request: AugmentedRequest) {
+    request.headers.authorization = this.token;
+  }
+
+  async getTraces(sessionId: string) {
+    return await this.get<FetchResponse<Trace[]>>(
+      `/api/session/${sessionId}/traces`
+    );
+  }
+
+  async getAllTraces() {
+    return await this.get<FetchResponse<Trace[]>>('/api/trace/');
+  }
+
+  async getSpans(traceId: string) {
+    return await this.get<FetchResponse<Span[]>>(`/api/trace/${traceId}/spans`);
+  }
+}

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/apps/graphql/src/index.ts

@@ -4,6 +4,7 @@ import { ApolloServer } from '@apollo/server';
 import { startStandaloneServer } from '@apollo/server/standalone';
 import gql from 'graphql-tag';
 import { AuthAPI } from './datasources/auth-api';
+import { DataParkAPI } from './datasources/datapark-api';
 import { resolvers } from './resolvers';
 
 const typeDefs = gql(
@@ -19,6 +20,7 @@ async function startApolloServer() {
       return Promise.resolve({
         dataSources: {
           authAPI: new AuthAPI({ token, cache }),
+          dataparkAPI: new DataParkAPI({ token, cache }),
         },
       });
     },

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/apps/graphql/src/resolvers.ts

@@ -8,13 +8,35 @@ import type { GraphQLError } from 'graphql';
 
 export const resolvers: Resolvers = {
   Query: {
+    me: async (_, _args, { dataSources }) => {
+      return (await dataSources.authAPI.getCurrentUser()).data;
+    },
     user: async (_, { id }, { dataSources }) => {
       return (await dataSources.authAPI.getUser(id)).data;
     },
+    api_keys: async (_, _args, { dataSources }) => {
+      return (await dataSources.authAPI.getAPIKeys()).data;
+    },
+    traces: async (_, { session_id }, { dataSources }) => {
+      return (await dataSources.dataparkAPI.getTraces(session_id)).data;
+    },
+    all_traces: async (_, _args, { dataSources }) => {
+      return (await dataSources.dataparkAPI.getAllTraces()).data;
+    },
+    spans: async (_, { trace_id }, { dataSources }) => {
+      return (await dataSources.dataparkAPI.getSpans(trace_id)).data;
+    },
   },
   Mutation: {
-    createAPIKey: async (_, _args, { dataSources }) => {
-      return await mutationAdaptor(dataSources.authAPI.createAPIKey());
+    createAPIKey: async (_, { name }, { dataSources }) => {
+      return await mutationAdaptor(
+        dataSources.authAPI.createAPIKey(name || 'Secret Key')
+      );
+    },
+    updateAPIKey: async (_, { id, name }, { dataSources }) => {
+      return await mutationAdaptor(
+        dataSources.authAPI.updateAPIKey(id, name || undefined)
+      );
     },
     revokeAPIKey: async (_, { id }, { dataSources }) => {
       return await mutationAdaptor(dataSources.authAPI.revokeAPIKey(id));
@@ -35,7 +57,9 @@ export const resolvers: Resolvers = {
       );
     },
     updateUser: async (_, { id, name }, { dataSources }) => {
-      return await mutationAdaptor(dataSources.authAPI.updateUser(id, name));
+      return await mutationAdaptor(
+        dataSources.authAPI.updateUser(id, name ?? undefined)
+      );
     },
     createUser: async (_, { email, password, username }, { dataSources }) => {
       return await mutationAdaptor(

{divi-0.0.1.dev13 → divi-0.0.1.dev14}/apps/graphql/src/schema.graphql

@@ -1,7 +1,29 @@
 "Query is a collection of queries that can be made to the API"
 type Query {
+  "Fetch current user"
+  me: User!
   "Fetch a specific user by id"
   user(id: ID!): User!
+  "Fetch current user's API keys"
+  api_keys: [APIKey!]
+  "Fetch traces by session id"
+  traces(session_id: ID!): [Trace!]
+  "Fetch all traces"
+  all_traces: [Trace!]
+  "Fetch all spans by trace id"
+  spans(trace_id: ID!): [Span!]
+}
+
+"NullTime is a custom scalar type that represents a time value that can be null"
+type NullTime {
+  Time: String!
+  Valid: Boolean!
+}
+
+"KeyValue is a key-value pair"
+type KeyValue {
+  key: String!
+  value: String!
 }
 
 "User is a registered user of the application"
@@ -10,13 +32,37 @@ type User {
   name: String
   username: String!
   email: String!
-  api_keys: [APIKey]
+  api_keys: [APIKey!]
 }
 
 "APIKey is a key used to authenticate requests to the API"
 type APIKey {
   id: ID!
   api_key: String!
+  name: String
+  created_at: String!
+}
+
+"Trace is a record to track the execution of a session"
+type Trace {
+  id: ID!
+  name: String
+  start_time: String!
+  end_time: NullTime!
+  session_id: ID!
+}
+
+"Span is a record of a single unit of work within a trace"
+type Span {
+  id: ID!
+  parent_id: ID!
+  trace_id: ID!
+  name: String!
+  start_time: String!
+  end_time: String!
+  duration: Int!
+  kind: String!
+  metadata: [KeyValue!]
 }
 
 "Mutation is a collection of mutations that can be made to the API"
@@ -25,7 +71,8 @@ type Mutation {
   login(identity: String!, password: String!): CreateTokenResponse!
   loginWithAPIKey(api_key: String!): CreateTokenResponse!
   "API Key Mutations"
-  createAPIKey: CreateAPIKeyResponse!
+  createAPIKey(name: String): CreateAPIKeyResponse!
+  updateAPIKey(id: ID!, name: String): UpdateAPIKeyResponse!
   revokeAPIKey(id: ID!): RevokeAPIKeyResponse!
   "User Mutations"
   createUser(
@@ -33,7 +80,7 @@ type Mutation {
     email: String!
     password: String!
   ): CreateUserResponse!
-  updateUser(id: ID!, name: String!): UpdateUserResponse!
+  updateUser(id: ID!, name: String): UpdateUserResponse!
   deleteUser(id: ID!, password: String!): DeleteUserResponse!
 }
 
@@ -52,6 +99,14 @@ type CreateAPIKeyResponse implements MutationResponse {
   data: APIKey
 }
 
+"UpdateAPIKeyResponse is a response to the updateAPIKey mutation"
+type UpdateAPIKeyResponse implements MutationResponse {
+  code: Int!
+  message: String!
+  success: Boolean!
+  data: APIKey
+}
+
 "RevokeAPIKeyResponse is a response to the revokeAPIKey mutation"
 type RevokeAPIKeyResponse implements MutationResponse {
   code: Int!

divi-0.0.1.dev14/apps/graphql/src/types/context.d.ts

@@ -0,0 +1,9 @@
+import type { AuthAPI } from '@/datasources/auth-api';
+import type { DataParkAPI } from '@/datasources/datapark-api';
+
+export type DataSourceContext = {
+  dataSources: {
+    authAPI: AuthAPI;
+    dataparkAPI: DataParkAPI;
+  };
+};