Merge pull request #23 from indexnetwork/indexer-authenticate

iterate
indexnetwork · Feb 7, 2024 · ebba487 · ebba487
2 parents 0535afb + fc690ee
commit ebba487
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/api/src/libs/kafka-indexer.js b/api/src/libs/kafka-indexer.js
@@ -2,7 +2,7 @@ import dotenv from 'dotenv'
 import axios from 'axios'
 import { ItemService } from "../services/item.js";
 import { EmbeddingService } from "../services/embedding.js";
-import { getPKPSession } from "../libs/lit/index.js";
+import { getPKPSession, getPKPSessionForIndexer } from "../libs/lit/index.js";
 
 if(process.env.NODE_ENV !== 'production'){
     dotenv.config()
@@ -12,11 +12,11 @@ export const createIndexItemEvent = async (id) => {
     console.log("createIndexItemEvent", id)
 
     const itemService = new ItemService()
-    const indexItem = await itemService.getIndexItemById(id);
+    const indexItem = await itemService.getIndexItemById(id, false);
 
     try {
 
-        const indexSession = await getPKPSession(indexItem.index);
+        const indexSession = await getPKPSessionForIndexer(indexItem.index);
         await indexSession.did.authenticate();
 
         const embeddingResponse = await axios.post(`${process.env.LLM_INDEXER_HOST}/embeddings`, {

diff --git a/api/src/libs/lit/index.js b/api/src/libs/lit/index.js
@@ -119,6 +119,14 @@ export const decodeDIDWithLit = (encodedDID) => {
 
 export const walletToDID = (chain, wallet) => `did:pkh:eip155:${parseInt(chain).toString()}:${wallet}`
 
+
+export const getPKPSessionForIndexer = async(index) => {
+  const session = await DIDSession.fromSession(process.env.INDEXER_DID_SESSION);
+  await session.did.authenticate()
+
+  const pkpSession =  await getPKPSession(session, index);
+  return pkpSession;
+}
 export const getPKPSession = async (session, index) => {
 
 	if(!session.did.authenticated){

diff --git a/api/src/services/item.js b/api/src/services/item.js
@@ -108,7 +108,7 @@ export class ItemService {
         }
     }
     // Todo make it multimodel later.
-    async getIndexItemById(indexItemId) {
+    async getIndexItemById(indexItemId, transformation=true) {
 
         try {
             const {data, errors} = await this.client.executeQuery(`
@@ -127,7 +127,7 @@ export class ItemService {
                 throw new Error('Invalid response data');
             }
 
-            return transformIndexItem(data.node);
+            return transformation ?transformIndexItem(data.node) : data.node;
 
         } catch (error) {
             // Log the error and rethrow it for external handling