Nia - AI code agent that actually understands your codebase

@john_tans thank you so much! lmk how it goes: https://discord.gg/8M3T3mgn8k

@bellam_saiteja Hey Saiteja, thanks for the congrats!

What really sets Nia AI apart is that it indexes your entire codebase, so it can handle multi-file queries and map dependencies across 100K+ lines of code—something the others can’t do with their limited token windows. Plus, we’re working on an upcoming “Agent Mode” that will automatically flag issues, suggest refactors, and even kick off code reviews and PRs on every commit. Personally, I struggle a lot with code reviews so having a teammate by my side would help a lot:)

Hope this helps. Lmk what you think!

@arlanrakh Great, how are you handling token windows while working with large codebases?

@bellam_saiteja Great question, Saiteja! We tackle token window limitations by dynamically chunking large codebases using chunking + vector DBs . Then, we use a multi-step retrieval pipeline with LangChain/LangGraph with other tools to help us manage very large codebases.

@arlanrakh what's the max context window that you can support? How do you achieve that?

@chrismessina Great question! We don't rely on a fixed max context window—instead, we dynamically break your codebase into smaller, vectorized chunks stored in a vector db. While the underlying LLM has a token limit (say, 200k tokens for 3.5 sonnet), our approach effectively lets you work with context spanning 100K+ lines of code.

@vladzima Just shipped a new update where Nia recaps the question asked earlier in the thread. It should be live in 5 minutes.

Example:

@Nia AI how does their recommendation algorithm works?

Response from Nia:

Question:
@Nia AI how does their recommendation algorithm works?

Sources:
• twitter/the-algorithm/README.md
• twitter/the-algorithm/follow-recommendations-service/README.md
• twitter/the-algorithm/follow-recommendations-service/README.md
• twitter/the-algorithm/product-mixer/core/src/main/scala/com/twitter/product_mixer/core/pipeline/recommendation/BUILD
• twitter/the-algorithm/follow-recommendations-service/server/src/main/scala/com/twitter/follow_recommendations/flows/post_nux_ml/BUILD
• twitter/the-algorithm/follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/base/RecommendationFlow.scala
• twitter/the-algorithm/follow-recommendations-service/README.md
• twitter/the-algorithm/src/scala/com/twitter/recos/user_tweet_entity_graph/RecommendationHandler.scala
• twitter/the-algorithm/src/scala/com/twitter/recos/user_user_graph/RecommendUsersHandler.scala
• twitter/the-algorithm/follow-recommendations-service/thrift/src/main/thrift/recommendations.thriftBased on the repository code, I'll explain how Twitter's recommendation algorithm works, primarily focusing on the Follow Recommendations Service (FRS) as it's one of the main components detailed in the codebase.From the main README.md and follow-recommendations-service documentation, here's how the recommendation system works:1. Overall Architecture
The recommendation algorithm is responsible for serving feeds of Tweets and other content across all Twitter product surfaces (e.g., For You Timeline, Search, Explore, Notifications).2. Follow Recommendations Process
The Follow Recommendations Service (FRS) works through several key steps:a) Candidate Generation
- Located in follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/candidate_sources/
- Uses various user signals and algorithms to identify potential accounts to recommend
- Includes methods like:
  - Triangular loops
  - Two-hop random walk
  - User-user graph analysisb) Filtering
- Located in follow-recommendations-service/common/src/main/scala/com/twitter/follow_recommendations/common/predicates
- Applies filtering logic to improve quality and health
- Happens both before and after ranking
- Heavier filtering (higher latency operations) typically occurs post-rankingc) Ranking
- Uses both Machine Learning (ML) and heuristic rule-based ranking
- Process:
  1. ML features are fetched (feature hydration)
  2. Creates DataRecord for each <user, candidate> pair
  3. Sends pairs to ML prediction service
  4. Returns prediction score based on:
     - Probability of follow given recommendation
     - Probability of positive engagement after follow
  5. Uses these scores to rank candidatesd) Transform
- Performs necessary transformations on candidates:
  - Deduplication
  - Attaches social proof (e.g., "followed by XX user")
  - Adds tracking tokense) Truncation
- Final step that trims the candidate pool to a specified size
- Ensures only the most relevant candidates are presented3. Social Proof and Recommendations
From the RecommendationHandler.scala, we can see that the system also handles:
- Tweet recommendations
- Social proof hydration
- Different types of recommendations (tweets, hashtags, URLs)The system is highly customizable per use case, with different display locations having their own specific flows and configurations. This allows Twitter to optimize recommendations based on where and how they're being displayed in the product.This is a high-level overview based on the provided repository code. The actual implementation involves many more intricate details and specialized algorithms for different recommendation scenarios.

@arlanrakh Awesome!

@sukhrob_abdullaev Thank you brother!

@cranqnow Thank you so much!

@michael_vavilov It is meant to be more of like context-aware programmer who is integrated deep within your codebase rather than any typical code editor on the market

@omar_dadabhoy2 Proud to have you! Thanks for your support brother!!

@shant_hambarsoumian Thank you so much! Let me know if you have any feedback! Always happy to help.

@alan_rivera Thank you so much! I am available 24/7, just DM me on Linkedin: www.linkedin.com/in/arlan-rakhmetzhanov

@hhendrix Always welcome! Keywords is such a game-changer.

@ohansemmanuel Hey, thanks for asking! While Cursor is great code editor overall, Nia AI is built to index your entire codebase. This means we overcome the 200k-token window limitation, enabling multi-file queries and a complete view of dependencies across 100K+ lines of code. We’re focused on delivering deep architecture insights and automating code reviews—plus, our upcoming Agent Mode will proactively suggest refactors and flag issues. In short, Nia AI is designed for larger, more complex projects where understanding the whole picture is key (+deep github integration).

@arlanrakh I see. I'll give a try then. Thanks!

@aleksbchkv Great question, and thanks for the kind words!

Here’s how we handle privacy:

When you upload your code, Nia processes it to generate vector embeddings and an internal context—this helps power our precise, multi-file insights. However, we don’t store your raw source code permanently. The embeddings we generate are ephemeral and isolated to your project only. This means that insights, best practices, or any contextual data derived from your code are not shared or used to influence recommendations for other customers. Each customer’s data is kept separate to ensure privacy and confidentiality.

Hope that clears things up!

@admiralrohan Hey, thanks for asking! While Cursor is great code editor overall, Nia AI is built to index your entire codebase. This means we overcome the 200k-token window limitation, enabling multi-file queries and a complete view of dependencies across 100K+ lines of code. We’re focused on delivering deep architecture insights and automating code reviews—plus, our upcoming Agent Mode will proactively suggest refactors and flag issues. In short, Nia AI is designed for larger, more complex projects where understanding the whole picture is key (+deep github integration).

2. There is no vscode extension but a dedicated webapp that you can find at app.trynia.ai

@arlanrakh I chatted with cal.com repo for testing and don't see any chat history in home page.
And it feels like a good way to understand new repository and onboarding new team members to new projects. I am not yet sure how can it help me with developing new features.

@kay_arkain thank you so much! lmk how it goes

@hussein_r Agreed! Thanks for your support!

@davidcardenas Thanks bro! Hope you will use it for cogent

@yuval_comunityapp Thank you so much! Lmk if you have any feedback - that means a lot!

@chad_carlson Thank you! Let me know if you encounter any bugs/issues: https://discord.gg/8M3T3mgn8k

@divyansh_tiwari7 Thank you for your support!